An extension of the nanoGPT repository for training small MOE models.
☆266Mar 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:
- Code in support of the paper Continuous Mixtures of Tractable Probabilistic Models☆12Oct 12, 2024Updated last year
- train with kittens!☆64Oct 25, 2024Updated last year
- Minimalistic large language model 3D-parallelism training☆2,663Apr 7, 2026Updated 3 weeks ago
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,159Aug 26, 2025Updated 8 months ago
- The CodeInsight dataset is designed for code generation tasks, providing developers with expert-curated examples that bridge the gap betw…☆14Oct 22, 2024Updated 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.
- NanoGPT (124M) in 90 seconds☆5,157Updated this week
- Vocabulary Parallelism☆26Mar 10, 2025Updated last year
- RLHF experiments on a single A100 40G GPU. Support PPO, GRPO, REINFORCE, RAFT, RLOO, ReMax, DeepSeek R1-Zero reproducing.☆80Feb 19, 2025Updated last year
- Language models scale reliably with over-training and on downstream tasks☆101Apr 2, 2024Updated 2 years ago
- image captioninggg🐳☆12Aug 30, 2024Updated last year
- [NAACL'25] "Revealing the Barriers of Language Agents in Planning"☆13Jun 22, 2025Updated 10 months ago
- 🚀 Efficient implementations for emerging model architectures☆4,999Updated this week
- ☆22Oct 22, 2025Updated 6 months ago
- KV Cache Steering for Inducing Reasoning in Small Language Models☆48Jul 24, 2025Updated 9 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- OLMoE: Open Mixture-of-Experts Language Models☆1,012Sep 23, 2025Updated 7 months ago
- nanoGPT-like codebase for LLM training☆117Nov 7, 2025Updated 5 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 7 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…☆285Nov 24, 2025Updated 5 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆192Jan 19, 2026Updated 3 months ago
- [ACL'24 Oral] Analysing The Impact of Sequence Composition on Language Model Pre-Training☆23Aug 18, 2024Updated last year
- Simple and efficient pytorch-native transformer training and inference (batched)☆78Apr 2, 2024Updated 2 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Homepage for ProLong (Princeton long-context language models) and paper "How to Train Long-Context Language Models (Effectively)"☆250Sep 12, 2025Updated 7 months ago
- Composition of Multimodal Language Models From Scratch☆15Aug 16, 2024Updated last year
- Website for the book "The Elements of Differentiable Programming".☆14Jul 2, 2025Updated 9 months ago
- A family of open-sourced Mixture-of-Experts (MoE) Large Language Models☆1,677Mar 8, 2024Updated 2 years ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆73Apr 22, 2025Updated last year
- Puzzles for learning Triton☆2,404Apr 1, 2026Updated last month
- Building Blocks for Equivariant Neural Networks in e3nn and PyTorch 2.0☆19Nov 16, 2025Updated 5 months ago
- torchlogic is a pytorch framework for developing Neuro-Symbolic AI systems and implements Neural Reasoning Networks.☆18Sep 18, 2025Updated 7 months ago
- List of papers on Self-Correction of LLMs.☆80Dec 28, 2024Updated last year
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.☆15Jan 3, 2023Updated 3 years ago
- An implementation of EinsumNetworks in PyTorch.☆23Jan 26, 2026Updated 3 months ago
- Tool for learning linear temporal logic formulas using combinatorial approach☆13Jan 11, 2025Updated last year
- FlexiTokens☆21Dec 27, 2025Updated 4 months ago
- nanoGRPO is a lightweight implementation of Group Relative Policy Optimization (GRPO)☆142May 8, 2025Updated 11 months ago
- Source code for Leveraging 2D Information for Long-term Time Series Forecasting with Vanilla Transformers☆19May 29, 2024Updated last year
- Zero Bubble Pipeline Parallelism☆452May 7, 2025Updated 11 months ago