srush / Autodiff-PuzzlesLinks
☆456Updated last year
Alternatives and similar repositories for Autodiff-Puzzles
Users that are interested in Autodiff-Puzzles are comparing it to the libraries listed below
Sorting:
- Puzzles for exploring transformers☆373Updated 2 years ago
- What would you do with 1000 H100s...☆1,113Updated last year
- ☆283Updated last year
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆671Updated this week
- An interactive exploration of Transformer programming.☆269Updated last year
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆400Updated this week
- 🧱 Modula software package☆291Updated 2 months ago
- For optimization algorithm research and development.☆543Updated this week
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆658Updated this week
- ☆541Updated last year
- Annotated version of the Mamba paper☆489Updated last year
- Solve puzzles. Learn CUDA.☆64Updated last year
- Deep learning for dummies. All the practical details and useful utilities that go into working with real models.☆820Updated 2 months ago
- A puzzle to learn about prompting☆135Updated 2 years ago
- ☆510Updated 2 months ago
- Compositional Linear Algebra☆489Updated 2 months ago
- Resources for skilling up in AI alignment research engineering. Covers basics of deep learning, mechanistic interpretability, and RL.☆229Updated 2 months ago
- Building blocks for foundation models.☆566Updated last year
- Universal Notation for Tensor Operations in Python.☆438Updated 6 months ago
- Resources from the EleutherAI Math Reading Group☆54Updated 7 months ago
- Library for reading and processing ML training data.☆570Updated this week
- Named Tensors for Legible Deep Learning in JAX☆210Updated last week
- Implementation of Diffusion Transformer (DiT) in JAX☆295Updated last year
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆295Updated last year
- An interactive HTML pretty-printer for machine learning research in IPython notebooks.☆448Updated 2 months ago
- Best practices & guides on how to write distributed pytorch training code☆517Updated this week
- Open-source framework for the research and development of foundation models.☆501Updated last week
- seqax = sequence modeling + JAX☆168Updated 3 months ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆180Updated 5 months ago
- JAX Synergistic Memory Inspector☆179Updated last year