srush / Autodiff-Puzzles
☆418Updated 4 months ago
Alternatives and similar repositories for Autodiff-Puzzles:
Users that are interested in Autodiff-Puzzles are comparing it to the libraries listed below
- Puzzles for exploring transformers☆334Updated last year
- What would you do with 1000 H100s...☆1,013Updated last year
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆554Updated this week
- ☆212Updated 7 months ago
- Compositional Linear Algebra☆464Updated last month
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆370Updated this week
- For optimization algorithm research and development.☆498Updated 2 weeks ago
- Annotated version of the Mamba paper☆474Updated last year
- An interactive exploration of Transformer programming.☆259Updated last year
- A puzzle to learn about prompting☆124Updated last year
- Implementation of Diffusion Transformer (DiT) in JAX☆267Updated 9 months ago
- Solve puzzles. Learn CUDA.☆63Updated last year
- Building blocks for foundation models.☆460Updated last year
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆359Updated last month
- A Jax-based library for designing and training transformer models from scratch.☆282Updated 6 months ago
- Puzzles for learning Triton☆1,494Updated 3 months ago
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆125Updated last year
- jax-triton contains integrations between JAX and OpenAI Triton☆382Updated this week
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆520Updated 3 weeks ago
- Library for reading and processing ML training data.☆401Updated this week
- 🧱 Modula software package☆169Updated this week
- Machine Learning with Symbolic Tensors☆262Updated last week
- ☆148Updated last year
- Named Tensors for Legible Deep Learning in JAX☆165Updated this week
- Everything you want to know about Google Cloud TPU☆517Updated 7 months ago
- Uncertainty quantification with PyTorch☆344Updated last week
- TensorDict is a pytorch dedicated tensor container.☆891Updated this week
- CLU lets you write beautiful training loops in JAX.☆335Updated last week