Simple & Scalable Pretraining for Neural Architecture Research
☆347Mar 31, 2026Updated 5 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 7 months ago
- Memory optimized Mixture of Experts☆84Jul 25, 2025Updated last year
- Dion optimizer algorithm☆542Updated this week
- The official repository for SkyLadder: Better and Faster Pretraining via Context Window Scheduling☆43Dec 29, 2025Updated 8 months ago
- Efficient PScan implementation in PyTorch☆17Jan 2, 2024Updated 2 years ago
- 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.
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆203Jan 19, 2026Updated 8 months ago
- ☆287Jun 6, 2025Updated last year
- Checkpoint-engine is a simple middleware to update model weights in LLM inference engines☆1,004Updated this week
- Physics of Language Models: Part 4.2, Canon Layers at Scale where Synthetic Pretraining Resonates in Reality☆361Aug 21, 2026Updated 3 weeks ago
- Kinetics: Rethinking Test-Time Scaling Laws☆87Jul 11, 2025Updated last year
- 🚀 Efficient implementations for emerging model architectures☆5,764Updated this week
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆19Mar 15, 2024Updated 2 years ago
- Minimalistic large language model 3D-parallelism training☆2,825Updated this week
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆17Jan 7, 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.
- ☆23Jun 12, 2025Updated last year
- Official repository for Parallax (Parameterized Local Linear Attention)☆69Jul 30, 2026Updated last month
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆33May 25, 2024Updated 2 years ago
- DeltaProduct is a new linear recurrent neural network architecture that uses products of generalized Householder matrices as state-transi…☆17Oct 13, 2025Updated 11 months ago
- ☆274Updated this week
- 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 last year
- Official PyTorch implementation and models for paper "Diffusion Beats Autoregressive in Data-Constrained Settings". We find diffusion mod…☆128Jan 10, 2026Updated 8 months ago
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆262Jun 15, 2025Updated last year
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Open-sourcing code associated with the AAAI-25 paper "On the Expressiveness and Length Generalization of Selective State-Space Models on …☆17Sep 18, 2025Updated last year
- Official implementation for DenseMixer: Improving MoE Post-Training with Precise Router Gradient☆68Aug 3, 2025Updated last year
- Spectral Sphere Optimizer☆133Mar 23, 2026Updated 5 months ago
- [NeurIPS 2025 Spotlight] Reasoning Environments for Reinforcement Learning with Verifiable Rewards☆1,509Apr 17, 2026Updated 5 months ago
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning☆154Feb 25, 2026Updated 6 months ago
- The Automated LLM Speedrunning Benchmark measures how well LLM agents can reproduce previous innovations and discover new ones in languag…☆146May 6, 2026Updated 4 months ago
- Muon is Scalable for LLM Training☆1,546Aug 3, 2025Updated last year
- A Quirky Assortment of CuTe Kernels☆1,146Updated this week
- H-Net: Hierarchical Network with Dynamic Chunking☆880Nov 20, 2025Updated 9 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.
- Code for ICLR 2025 Paper "What is Wrong with Perplexity for Long-context Language Modeling?"☆118Oct 11, 2025Updated 11 months ago
- Official Code Repository for the paper "Key-value memory in the brain"☆34Feb 25, 2025Updated last year
- Our library for RL environments + evals☆4,631Updated this week
- ☆93Aug 18, 2024Updated 2 years ago
- 🔥 A minimal training framework for scaling FLA models☆416Apr 22, 2026Updated 4 months ago
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆135Jun 24, 2025Updated last year
- Accelerating MoE with IO and Tile-aware Optimizations☆769Aug 29, 2026Updated 3 weeks ago