An extension of the nanoGPT repository for training small MOE models.
☆276Mar 9, 2025Updated last year
Alternatives and similar repositories for nanoMoE
Users that are interested in nanoMoE are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- train with kittens!☆66Oct 25, 2024Updated last year
- Minimalistic large language model 3D-parallelism training☆2,729May 26, 2026Updated last month
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,228Aug 26, 2025Updated 10 months ago
- NanoGPT (124M) in 90 seconds☆5,438Jun 21, 2026Updated last week
- RLHF experiments on a single A100 40G GPU. Support PPO, GRPO, REINFORCE, RAFT, RLOO, ReMax, DeepSeek R1-Zero reproducing.☆80Feb 19, 2025Updated last year
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆92Jul 17, 2025Updated 11 months ago
- Language models scale reliably with over-training and on downstream tasks☆101Apr 2, 2024Updated 2 years ago
- DPO, but faster 🚀☆52Dec 6, 2024Updated last year
- image captioninggg🐳☆12Aug 30, 2024Updated last year
- [NAACL'25] "Revealing the Barriers of Language Agents in Planning"☆13Jun 22, 2025Updated last year
- Node based programming tool☆12Jan 8, 2023Updated 3 years ago
- 🚀 Efficient implementations for emerging model architectures☆5,249Jun 23, 2026Updated last week
- A collection of commonly used datasets as benchmarks for density estimation in MaLe☆20Jul 15, 2019Updated 6 years ago
- ☆18Dec 31, 2024Updated last year
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- KV Cache Steering for Inducing Reasoning in Small Language Models☆49Jul 24, 2025Updated 11 months ago
- OLMoE: Open Mixture-of-Experts Language Models☆1,031Sep 23, 2025Updated 9 months ago
- Local DeepSearch (Advantage: Low Threshold): an implementation of Agentic RAG based on DeepSeek-R1 API and Tavily API☆17Jun 21, 2025Updated last year
- nanoGPT-like codebase for LLM training☆118Nov 7, 2025Updated 7 months ago
- This is a simple torch implementation of the high performance Multi-Query Attention☆16Aug 23, 2023Updated 2 years ago
- Sound Separation, Omni modal☆28Sep 15, 2025Updated 9 months ago
- ☆17Apr 7, 2025Updated last year
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆288Nov 24, 2025Updated 7 months ago
- [ACL'24 Oral] Analysing The Impact of Sequence Composition on Language Model Pre-Training☆23Aug 18, 2024Updated last year
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆197Jan 19, 2026Updated 5 months ago
- Simple and efficient pytorch-native transformer training and inference (batched)☆78Apr 2, 2024Updated 2 years ago
- Homepage for ProLong (Princeton long-context language models) and paper "How to Train Long-Context Language Models (Effectively)"☆255Sep 12, 2025Updated 9 months ago
- Jupyter Notebooks from book UNDERSTANDING DEEP LEARNING (Prof Simon Prince) that I could solve.☆17Mar 20, 2024Updated 2 years ago
- Composition of Multimodal Language Models From Scratch☆15Aug 16, 2024Updated last year
- PyMolAI provides Agentic interface to PyMol workflows, built on top of Open Source version of PyMol.☆160Feb 23, 2026Updated 4 months ago
- CLIP is an open source, multimodal computer vision model and it's awesome!☆17Dec 16, 2024Updated last year
- NanoGPT-speedrunning for the poor T4 enjoyers☆74Apr 22, 2025Updated last year
- A family of open-sourced Mixture-of-Experts (MoE) Large Language Models☆1,692Mar 8, 2024Updated 2 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Building Blocks for Equivariant Neural Networks in e3nn and PyTorch 2.0☆19Nov 16, 2025Updated 7 months ago
- Puzzles for learning Triton☆2,499Apr 1, 2026Updated 3 months ago
- torchlogic is a pytorch framework for developing Neuro-Symbolic AI systems and implements Neural Reasoning Networks.☆21Sep 18, 2025Updated 9 months ago
- List of papers on Self-Correction of LLMs.☆81May 19, 2026Updated last month
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.☆14Jan 3, 2023Updated 3 years ago
- FlexiTokens☆23Dec 27, 2025Updated 6 months ago
- nanoGRPO is a lightweight implementation of Group Relative Policy Optimization (GRPO)☆144May 8, 2025Updated last year