lerax

An Open Source Machine Learning Framework for Everyone

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

Tensors and Dynamic neural networks in Python with strong GPU acceleration

??? 60+ Implementations/tutorials of deep learning papers with side-by-side notes ?; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), gans(cyclegan, stylegan2, ...), ? reinforcement learning (ppo, dqn), capsnet, distillation, ... ?

Deep Learning for humans

Deepfakes Software For All

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

? Industrial-strength Natural Language Processing (NLP) in Python

PyTorch Tutorial for Deep Learning Researchers

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