☆497Oct 18, 2024Updated 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☆386May 4, 2023Updated 2 years ago
- What would you do with 1000 H100s...☆1,154Jan 10, 2024Updated 2 years ago
- A puzzle to learn about prompting☆135May 12, 2023Updated 2 years ago
- Solve puzzles. Improve your pytorch.☆3,950Jul 15, 2024Updated last year
- Solve puzzles. Learn CUDA.☆11,970Sep 1, 2024Updated last year
- Puzzles for learning Triton☆2,314Nov 18, 2024Updated last year
- An interactive exploration of Transformer programming.☆271Nov 15, 2023Updated 2 years ago
- ☆22Apr 22, 2024Updated last year
- Experiment of using Tangent to autodiff triton☆82Jan 22, 2024Updated 2 years ago
- ☆294Updated this week
- A bibliography and survey of the papers surrounding o1☆1,212Nov 16, 2024Updated last year
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,869Jun 22, 2025Updated 8 months ago
- Annotated version of the Mamba paper☆497Feb 27, 2024Updated 2 years ago
- GPU programming related news and material links☆1,997Sep 17, 2025Updated 5 months ago
- Silly twitter torch implementations.☆46Oct 14, 2022Updated 3 years ago
- Solve puzzles. Learn CUDA.☆62Dec 13, 2023Updated 2 years ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆695Jan 26, 2026Updated last month
- Introductory lecture on Pytorch☆17May 8, 2022Updated 3 years ago
- Implementation of https://srush.github.io/annotated-s4☆512Jun 20, 2025Updated 8 months ago
- Named Tensors for Legible Deep Learning in JAX☆217Nov 8, 2025Updated 3 months ago
- JAX - A curated list of resources https://github.com/google/jax☆2,058Jan 20, 2026Updated last month
- Accelerated First Order Parallel Associative Scan☆194Jan 7, 2026Updated last month
- Understand and test language model architectures on synthetic tasks.☆254Feb 24, 2026Updated last week
- ☆566Jul 11, 2024Updated last year
- A PyTorch native platform for training generative AI models☆5,098Updated this week
- Minimalistic large language model 3D-parallelism training☆2,579Feb 19, 2026Updated last week
- JAX/Flax implementation of the Hyena Hierarchy☆34Apr 27, 2023Updated 2 years ago
- ☆109Jun 25, 2024Updated last year
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆863Updated this week
- ☆292Jul 15, 2024Updated last year
- ☆167Jul 5, 2023Updated 2 years ago
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,090Aug 26, 2025Updated 6 months ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆595Aug 12, 2025Updated 6 months ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,401Feb 20, 2026Updated last week
- Parallel Associative Scan for Language Models☆18Jan 8, 2024Updated 2 years ago
- Tile primitives for speedy kernels☆3,183Updated this week
- jax-triton contains integrations between JAX and OpenAI Triton☆439Updated this week
- Deep learning for dummies. All the practical details and useful utilities that go into working with real models.☆831Updated this week
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,743Feb 16, 2026Updated 2 weeks ago