Simple MPI implementation for prototyping or learning
☆315Aug 6, 2025Updated 9 months ago
Alternatives and similar repositories for nanoMPI
Users that are interested in nanoMPI are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Custom triton kernels for training Karpathy's nanoGPT.☆19Oct 21, 2024Updated last year
- Implementing DeepSeek R1's GRPO algorithm from scratch☆1,856Apr 18, 2025Updated last year
- Simple Byte pair Encoding mechanism used for tokenization process . written purely in C☆147Nov 11, 2024Updated last year
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,188Aug 26, 2025Updated 9 months ago
- CIFAR-10 speedrun: Trains to 94% accuracy in 1.98 seconds on a single NVIDIA A100 GPU.☆76Oct 17, 2025Updated 7 months ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.☆347Nov 2, 2025Updated 6 months ago
- ☆22May 5, 2025Updated last year
- workflow of nndeploy☆13Nov 5, 2025Updated 6 months ago
- An experimental communicating attention kernel based on DeepEP.☆34Jul 29, 2025Updated 10 months ago
- The simplest, fastest repository for training/finetuning small-sized VLMs.☆4,876Oct 27, 2025Updated 7 months ago
- UNet diffusion model in pure CUDA☆657Jun 28, 2024Updated last year
- Tile primitives for speedy kernels☆3,377May 22, 2026Updated last week
- ☆10Sep 30, 2015Updated 10 years ago
- ☆33Apr 19, 2025Updated last year
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆600May 13, 2026Updated 2 weeks ago
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,574Jan 12, 2025Updated last year
- Industrial Language-Image Dataset (ILID), a web-crawled dataset containing language-image samples from various web catalogs, representing…☆17Jun 4, 2025Updated 11 months ago
- ☆57Feb 24, 2026Updated 3 months ago
- Learn CUDA with PyTorch☆312May 24, 2026Updated last week
- Efficient Triton Kernels for LLM Training☆6,393Updated this week
- Implementation of Strassen attention, from Kozachinskiy et al. of National Center of AI in Chile☆41Jul 8, 2025Updated 10 months ago
- Fast bare-bones BPE for modern tokenizer training☆178Jun 23, 2025Updated 11 months ago
- A PyTorch native platform for training generative AI models☆5,391Updated this week
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Tilus is a tile-level kernel programming language with explicit control over shared memory and registers.☆486May 19, 2026Updated last week
- Simple Transformer in Jax☆143Jun 22, 2024Updated last year
- ☆93Jul 5, 2024Updated last year
- RuLES: a benchmark for evaluating rule-following in language models☆253Feb 24, 2025Updated last year
- Explorations into the proposed Streaming Deep Reinforcement Learning, from University of Alberta☆30May 18, 2026Updated last week
- ☆25May 23, 2025Updated last year
- ☆19Apr 22, 2025Updated last year
- The Automated LLM Speedrunning Benchmark measures how well LLM agents can reproduce previous innovations and discover new ones in languag…☆142May 6, 2026Updated 3 weeks ago
- gpt-2 from scratch in mlx☆428Jun 12, 2024Updated last year
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Minimal yet performant LLM examples in pure JAX☆257Apr 10, 2026Updated last month
- A numeric optimization package for Torch.☆41Aug 19, 2021Updated 4 years ago
- ring-attention experiments☆167Oct 17, 2024Updated last year
- Where GPUs get cooked 👩🍳🔥☆393Updated this week
- Implementation of Diffusion Transformer (DiT) in JAX☆313Jun 11, 2024Updated last year
- ☆18Dec 2, 2024Updated last year
- Code for the papers: “Four Over Six: More Accurate NVFP4 Quantization with Adaptive Block Scaling” and “Adaptive Block-Scaled Data Types”☆187Apr 21, 2026Updated last month