divyamakkar0 / JAXformerLinks
A zero-to-one guide on scaling modern transformers with n-dimensional parallelism.
☆104Updated 2 months ago
Alternatives and similar repositories for JAXformer
Users that are interested in JAXformer are comparing it to the libraries listed below
Sorting:
- MoE training for Me and You and maybe other people☆239Updated last week
- Simple Transformer in Jax☆141Updated last year
- NanoGPT-speedrunning for the poor T4 enjoyers☆73Updated 8 months ago
- ☆286Updated last year
- Minimal yet performant LLM examples in pure JAX☆219Updated 3 weeks ago
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆195Updated 6 months ago
- ☆115Updated 2 weeks ago
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆108Updated 9 months ago
- seqax = sequence modeling + JAX☆169Updated 5 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆180Updated 5 months ago
- SIMD quantization kernels☆93Updated 3 months ago
- Dion optimizer algorithm☆409Updated this week
- ☆91Updated last year
- Custom triton kernels for training Karpathy's nanoGPT.☆19Updated last year
- Compiling useful links, papers, benchmarks, ideas, etc.☆45Updated 9 months ago
- 🧱 Modula software package☆316Updated 4 months ago
- ☆28Updated last year
- Storing long contexts in tiny caches with self-study☆226Updated 2 weeks ago
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆153Updated 2 years ago
- ☆225Updated last month
- Quantized LLM training in pure CUDA/C++.☆224Updated this week
- Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*☆87Updated 2 years ago
- The Automated LLM Speedrunning Benchmark measures how well LLM agents can reproduce previous innovations and discover new ones in languag…☆121Updated 2 months ago
- FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.☆327Updated last month
- A puzzle to learn about prompting☆135Updated 2 years ago
- A MAD laboratory to improve AI architecture designs 🧪☆135Updated last year
- DeMo: Decoupled Momentum Optimization☆197Updated last year
- PTX-Tutorial Written Purely By AIs (Deep Research of Openai and Claude 3.7)☆66Updated 9 months ago
- Write a fast kernel and run it on Discord. See how you compare against the best!☆64Updated this week
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆132Updated last year