Simple MPI implementation for prototyping or learning
☆304Aug 6, 2025Updated 7 months ago
Alternatives and similar repositories for nanoMPI
Users that are interested in nanoMPI are comparing it to the libraries listed below
Sorting:
- Custom triton kernels for training Karpathy's nanoGPT.☆19Oct 21, 2024Updated last year
- CIFAR-10 speedrun: Trains to 94% accuracy in 1.98 seconds on a single NVIDIA A100 GPU.☆56Oct 17, 2025Updated 4 months ago
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,099Aug 26, 2025Updated 6 months ago
- Implementing DeepSeek R1's GRPO algorithm from scratch☆1,781Apr 18, 2025Updated 10 months ago
- FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.☆336Nov 2, 2025Updated 4 months ago
- An experimental communicating attention kernel based on DeepEP.☆35Jul 29, 2025Updated 7 months ago
- ☆31Apr 19, 2025Updated 10 months ago
- ☆22May 5, 2025Updated 10 months ago
- Tile primitives for speedy kernels☆3,202Feb 24, 2026Updated last week
- ☆53Feb 24, 2026Updated last week
- The simplest, fastest repository for training/finetuning small-sized VLMs.☆4,685Oct 27, 2025Updated 4 months ago
- Code for "What really matters in matrix-whitening optimizers?"☆22Oct 31, 2025Updated 4 months ago
- Pytorch routines for (Ker)nel (Mac)hines☆11Oct 10, 2025Updated 4 months ago
- See https://github.com/cuda-mode/triton-index/ instead!☆11May 8, 2024Updated last year
- ☆12Aug 26, 2025Updated 6 months ago
- workflow of nndeploy☆13Nov 5, 2025Updated 4 months ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆595Aug 12, 2025Updated 6 months ago
- Simple Transformer in Jax☆143Jun 22, 2024Updated last year
- A selective knowledge distillation algorithm for efficient speculative decoders☆36Nov 27, 2025Updated 3 months ago
- ☆18Nov 11, 2025Updated 3 months ago
- ring-attention experiments☆166Oct 17, 2024Updated last year
- The Automated LLM Speedrunning Benchmark measures how well LLM agents can reproduce previous innovations and discover new ones in languag…☆133Feb 21, 2026Updated last week
- Learn CUDA with PyTorch☆231Feb 23, 2026Updated last week
- Efficient Triton Kernels for LLM Training☆6,189Updated this week
- Code for the paper “Four Over Six: More Accurate NVFP4 Quantization with Adaptive Block Scaling”☆130Updated this week
- Cloud modelling (Manchester Prize)☆15Apr 25, 2025Updated 10 months ago
- Minimal yet performant LLM examples in pure JAX☆240Jan 14, 2026Updated last month
- CHAI is a library for dynamic pruning of attention heads for efficient LLM inference.☆22Dec 11, 2024Updated last year
- Flash Attention from Scratch on CUDA Ampere☆146Sep 1, 2025Updated 6 months ago
- ☆15Mar 30, 2023Updated 2 years ago
- Implementation of Diffusion Transformer (DiT) in JAX☆306Jun 11, 2024Updated last year
- A bunch of kernels that might make stuff slower 😉☆75Feb 18, 2026Updated 2 weeks ago
- ☆65Apr 26, 2025Updated 10 months ago
- Tilus is a tile-level kernel programming language with explicit control over shared memory and registers.☆446Feb 4, 2026Updated last month
- ☆46May 24, 2025Updated 9 months ago
- ☆18Dec 2, 2024Updated last year
- ☆25May 23, 2025Updated 9 months ago
- A PyTorch native platform for training generative AI models☆5,098Feb 28, 2026Updated last week
- DeeperGEMM: crazy optimized version☆74May 5, 2025Updated 10 months ago