Simple & Scalable Pretraining for Neural Architecture Research
☆336Mar 31, 2026Updated 3 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:
- ☆126Feb 19, 2026Updated 5 months ago
- Memory optimized Mixture of Experts☆80Jul 25, 2025Updated 11 months ago
- Dion optimizer algorithm☆494Jul 12, 2026Updated last week
- The official repository for SkyLadder: Better and Faster Pretraining via Context Window Scheduling☆43Dec 29, 2025Updated 6 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.☆199Jan 19, 2026Updated 6 months ago
- ☆284Jun 6, 2025Updated last year
- Checkpoint-engine is a simple middleware to update model weights in LLM inference engines☆970Jul 4, 2026Updated 2 weeks ago
- Physics of Language Models: Part 4.2, Canon Layers at Scale where Synthetic Pretraining Resonates in Reality☆356May 20, 2026Updated 2 months ago
- Kinetics: Rethinking Test-Time Scaling Laws☆87Jul 11, 2025Updated last year
- 🚀 Efficient implementations for emerging model architectures☆5,379Updated this week
- Code for ICLR 2025 Paper "What is Wrong with Perplexity for Long-context Language Modeling?"☆115Oct 11, 2025Updated 9 months ago
- ☆21Jun 12, 2025Updated last year
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆17Jan 7, 2025Updated last year
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Minimalistic large language model 3D-parallelism training☆2,755May 26, 2026Updated last month
- Official repository for Parallax (Parameterized Local Linear Attention)☆65Jul 7, 2026Updated 2 weeks ago
- An MLX implementation of Meta AI's ESM-2 protein language model☆16Aug 16, 2025Updated 11 months ago
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆33May 25, 2024Updated 2 years ago
- ☆209Updated 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
- Official PyTorch implementation and models for paper "Diffusion Beats Autoregressive in Data-Constrained Settings". We find diffusion mod…☆127Jan 10, 2026Updated 6 months ago
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆257Jun 15, 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.
- Official implementation for DenseMixer: Improving MoE Post-Training with Precise Router Gradient☆67Aug 3, 2025Updated 11 months ago
- 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
- [NeurIPS 2025 Spotlight] Reasoning Environments for Reinforcement Learning with Verifiable Rewards☆1,463Apr 17, 2026Updated 3 months ago
- Spectral Sphere Optimizer☆130Mar 23, 2026Updated 3 months ago
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning☆150Feb 25, 2026Updated 4 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 2 months ago
- Muon is Scalable for LLM Training☆1,507Aug 3, 2025Updated 11 months ago
- A Quirky Assortment of CuTe Kernels☆1,063Updated this week
- H-Net: Hierarchical Network with Dynamic Chunking☆867Nov 20, 2025Updated 8 months ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Official Code Repository for the paper "Key-value memory in the brain"☆32Feb 25, 2025Updated last year
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆133Jun 24, 2025Updated last year
- ☆93Aug 18, 2024Updated last year
- 🔥 A minimal training framework for scaling FLA models☆403Apr 22, 2026Updated 2 months ago
- Accelerating MoE with IO and Tile-aware Optimizations☆732Jul 4, 2026Updated 2 weeks ago
- Official repository Flash Local Linear Attention☆37May 28, 2026Updated last month
- Our library for RL environments + evals☆4,389Updated this week