agjax

? Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Deep Learning for humans

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

? Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.

Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.

? A ranked list of awesome machine learning Python libraries. Updated weekly.

PyTorch tutorials and fun projects including neural talk, neural style, poem writing, anime generation (《深度学习框架PyTorch：入门与实战》)

The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)