KellerJordan / modded-nanogpt
NanoGPT (124M) quality in 7.8 8xH100-minutes
☆1,033Updated this week
Related projects ⓘ
Alternatives and complementary repositories for modded-nanogpt
- nanoGPT style version of Llama 3.1☆1,246Updated 3 months ago
- UNet diffusion model in pure CUDA☆584Updated 4 months ago
- The Multilayer Perceptron Language Model☆523Updated 3 months ago
- Official implementation of "Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modeling"☆803Updated 3 months ago
- Deep learning for dummies. All the practical details and useful utilities that go into working with real models.☆715Updated last month
- The Autograd Engine☆534Updated 2 months ago
- A native PyTorch Library for large model training☆2,623Updated this week
- System 2 Reasoning Link Collection☆693Updated 3 weeks ago
- The Tensor (or Array)☆411Updated 3 months ago
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,679Updated this week
- Schedule-Free Optimization in PyTorch☆1,898Updated 2 weeks ago
- Open weights language model from Google DeepMind, based on Griffin.☆607Updated 4 months ago
- The n-gram Language Model☆1,342Updated 3 months ago
- Puzzles for learning Triton☆1,135Updated this week
- Minimalistic large language model 3D-parallelism training☆1,260Updated this week
- Tile primitives for speedy kernels☆1,658Updated this week
- Best practices & guides on how to write distributed pytorch training code☆286Updated 2 weeks ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆483Updated 3 weeks ago
- Implementation of Diffusion Transformer (DiT) in JAX☆252Updated 5 months ago
- What would you do with 1000 H100s...☆903Updated 10 months ago
- Fast bare-bones BPE for modern tokenizer training☆142Updated last month
- Official repository for the paper "Grokfast: Accelerated Grokking by Amplifying Slow Gradients"☆515Updated 4 months ago
- Make PyTorch models up to 40% faster! Thunder is a source to source compiler for PyTorch. It enables using different hardware executors a…☆1,199Updated this week
- Annotated version of the Mamba paper☆457Updated 8 months ago
- For optimization algorithm research and development.☆449Updated this week
- ReFT: Representation Finetuning for Language Models☆1,159Updated 2 weeks ago
- A bibliography and survey of the papers surrounding o1☆754Updated this week
- ☆935Updated 2 weeks ago
- A benchmark to evaluate language models on questions I've previously asked them to solve.☆916Updated 2 weeks ago
- Recipes for shrinking, optimizing, customizing cutting edge vision models. 💜☆890Updated 2 months ago