Simple & Scalable Pretraining for Neural Architecture Research
☆341Mar 31, 2026Updated 4 months ago
Alternatives and similar repositories for ArchScale
Users that are interested in ArchScale are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆125Feb 19, 2026Updated 5 months ago
- Memory optimized Mixture of Experts☆81Jul 25, 2025Updated last year
- Dion optimizer algorithm☆498Updated this week
- The official repository for SkyLadder: Better and Faster Pretraining via Context Window Scheduling☆43Dec 29, 2025Updated 7 months ago
- Efficient PScan implementation in PyTorch☆17Jan 2, 2024Updated 2 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆201Jan 19, 2026Updated 6 months ago
- ☆285Jun 6, 2025Updated last year
- Checkpoint-engine is a simple middleware to update model weights in LLM inference engines☆993Updated this week
- Physics of Language Models: Part 4.2, Canon Layers at Scale where Synthetic Pretraining Resonates in Reality☆357May 20, 2026Updated 2 months ago
- 🚀 Efficient implementations for emerging model architectures☆5,528Updated this week
- Kinetics: Rethinking Test-Time Scaling Laws☆87Jul 11, 2025Updated last year
- Minimalistic large language model 3D-parallelism training☆2,779May 26, 2026Updated 2 months ago
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆17Jan 7, 2025Updated last year
- ☆21Jun 12, 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.
- Official repository for Parallax (Parameterized Local Linear Attention)☆68Jul 30, 2026Updated last week
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆33May 25, 2024Updated 2 years ago
- ☆253Updated this week
- DeltaProduct is a new linear recurrent neural network architecture that uses products of generalized Householder matrices as state-transi…☆15Oct 13, 2025Updated 9 months ago
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆92Jul 17, 2025Updated last year
- An MLX implementation of Meta AI's ESM-2 protein language model☆16Aug 16, 2025Updated 11 months ago
- Official PyTorch implementation and models for paper "Diffusion Beats Autoregressive in Data-Constrained Settings". We find diffusion mod…☆127Jan 10, 2026Updated 7 months ago
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆259Jun 15, 2025Updated last year
- Open-sourcing code associated with the AAAI-25 paper "On the Expressiveness and Length Generalization of Selective State-Space Models on …☆16Sep 18, 2025Updated 10 months 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.
- Official implementation for DenseMixer: Improving MoE Post-Training with Precise Router Gradient☆68Aug 3, 2025Updated last year
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning☆151Feb 25, 2026Updated 5 months ago
- Spectral Sphere Optimizer☆132Mar 23, 2026Updated 4 months ago
- [NeurIPS 2025 Spotlight] Reasoning Environments for Reinforcement Learning with Verifiable Rewards☆1,477Apr 17, 2026Updated 3 months ago
- The Automated LLM Speedrunning Benchmark measures how well LLM agents can reproduce previous innovations and discover new ones in languag…☆145May 6, 2026Updated 3 months ago
- Muon is Scalable for LLM Training☆1,528Aug 3, 2025Updated last year
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆18Mar 15, 2024Updated 2 years ago
- A Quirky Assortment of CuTe Kernels☆1,096Updated this week
- H-Net: Hierarchical Network with Dynamic Chunking☆871Nov 20, 2025Updated 8 months ago
- 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.
- Code for ICLR 2025 Paper "What is Wrong with Perplexity for Long-context Language Modeling?"☆116Oct 11, 2025Updated 9 months ago
- Official Code Repository for the paper "Key-value memory in the brain"☆32Feb 25, 2025Updated last year
- Our library for RL environments + evals☆4,479Updated this week
- ☆93Aug 18, 2024Updated last year
- 🔥 A minimal training framework for scaling FLA models☆409Apr 22, 2026Updated 3 months ago
- Accelerating MoE with IO and Tile-aware Optimizations☆737Jul 4, 2026Updated last month
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆134Jun 24, 2025Updated last year