Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
☆132Apr 17, 2024Updated 2 years ago
Alternatives and similar repositories for min-max-gpt
Users that are interested in min-max-gpt are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆95Jul 5, 2024Updated 2 years ago
- ☆26May 3, 2024Updated 2 years ago
- ☆24Jun 4, 2024Updated 2 years ago
- Parallel Associative Scan for Language Models☆18Jan 8, 2024Updated 2 years ago
- Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam☆89Jul 28, 2024Updated 2 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- ☆30Dec 2, 2024Updated last year
- WIP☆97Aug 13, 2024Updated 2 years ago
- ☆40Apr 27, 2024Updated 2 years ago
- Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun☆56Mar 10, 2025Updated last year
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆644Jul 1, 2024Updated 2 years ago
- These papers will provide unique insightful concepts that will broaden your perspective on neural networks and deep learning☆48Sep 3, 2023Updated 3 years ago
- ☆53Jan 6, 2024Updated 2 years ago
- Minimal Implementation of a D3PM in pytorch☆310Apr 22, 2024Updated 2 years ago
- ☆126May 28, 2024Updated 2 years ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- A library for unit scaling in PyTorch☆137Jul 11, 2025Updated last year
- supporting pytorch FSDP for optimizers☆85Dec 8, 2024Updated last year
- ☆20May 29, 2026Updated 5 months ago
- Efficient PScan implementation in PyTorch☆17Jan 2, 2024Updated 2 years ago
- Fine-Tuning Pre-trained Transformers into Decaying Fast Weights☆20Oct 9, 2022Updated 4 years ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆605Aug 14, 2026Updated 2 months ago
- Simple (fast) transformer inference in PyTorch with torch.compile + lit-llama code☆11Aug 29, 2023Updated 3 years ago
- Explorations into the recently proposed Taylor Series Linear Attention☆102Aug 27, 2026Updated 2 months ago
- Code for the paper "Function-Space Learning Rates"☆23Jun 3, 2025Updated last year
- 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.
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆19Mar 15, 2024Updated 2 years ago
- ☆33Sep 10, 2024Updated 2 years ago
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆97Dec 5, 2024Updated last year
- Minimalistic, extremely fast, and hackable researcher's toolbench for GPT models in 307 lines of code. Reaches <3.8 validation loss on wi…☆361Jul 29, 2024Updated 2 years ago
- ☆307Jul 15, 2024Updated 2 years ago
- Minimal (truly) muP implementation, consistent with TP4 and TP5 papers notation☆14Jan 2, 2026Updated 10 months ago
- ☆13Jun 3, 2024Updated 2 years ago
- Load & manage evolving datasets efficiently☆23Aug 22, 2025Updated last year
- ☆36Nov 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.
- ArXiV Notification Bot which sends you an email with the latest updates!☆17Oct 20, 2023Updated 3 years ago
- HGRN2: Gated Linear RNNs with State Expansion☆59Aug 20, 2024Updated 2 years ago
- ☆33Jan 6, 2025Updated last year
- Official implementation for "Q-Ensemble for Offline RL: Don't Scale the Ensemble, Scale the Batch Size", NeurIPS 2022, Offline RL Worksho…☆21Feb 27, 2023Updated 3 years ago
- seqax = sequence modeling + JAX☆199Jul 23, 2025Updated last year
- Understand and test language model architectures on synthetic tasks.☆291Mar 22, 2026Updated 7 months ago
- Official implementation of the paper "Linear Transformers with Learnable Kernel Functions are Better In-Context Models"☆168Jan 16, 2025Updated last year