srush / Autodiff-PuzzlesLinks
☆431Updated 7 months ago
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☆347Updated 2 years ago
- What would you do with 1000 H100s...☆1,048Updated last year
- ☆262Updated 10 months ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆584Updated this week
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆381Updated this week
- A puzzle to learn about prompting☆127Updated 2 years ago
- For optimization algorithm research and development.☆518Updated this week
- An interactive exploration of Transformer programming.☆264Updated last year
- Deep learning for dummies. All the practical details and useful utilities that go into working with real models.☆793Updated last month
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆370Updated last month
- Compositional Linear Algebra☆474Updated this week
- Building blocks for foundation models.☆500Updated last year
- Solve puzzles. Learn CUDA.☆64Updated last year
- 🧱 Modula software package☆194Updated 2 months ago
- Puzzles for learning Triton☆1,658Updated 6 months ago
- seqax = sequence modeling + JAX☆155Updated last month
- ☆474Updated 10 months ago
- jax-triton contains integrations between JAX and OpenAI Triton☆392Updated this week
- Annotated version of the Mamba paper☆482Updated last year
- Resources for skilling up in AI alignment research engineering. Covers basics of deep learning, mechanistic interpretability, and RL.☆213Updated last year
- TensorDict is a pytorch dedicated tensor container.☆925Updated this week
- Resources from the EleutherAI Math Reading Group☆53Updated 3 months ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆370Updated last month
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆287Updated 9 months ago
- An interactive HTML pretty-printer for machine learning research in IPython notebooks.☆416Updated 3 weeks ago
- An implementation of the transformer architecture onto an Nvidia CUDA kernel☆183Updated last year
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆544Updated this week
- Implementation of Diffusion Transformer (DiT) in JAX☆276Updated 11 months ago
- Named tensors with first-class dimensions for PyTorch☆329Updated last year
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆133Updated last year