KellerJordan / cifar10-airbench
94% on CIFAR-10 in 2.6 seconds π¨ 96% in 27 seconds
β195Updated last month
Alternatives and similar repositories for cifar10-airbench:
Users that are interested in cifar10-airbench are comparing it to the libraries listed below
- β146Updated last month
- supporting pytorch FSDP for optimizersβ75Updated last month
- 𧱠Modula software packageβ132Updated this week
- Efficient optimizersβ144Updated this week
- The AdEMAMix Optimizer: Better, Faster, Older.β178Updated 4 months ago
- Muon optimizer for neural networks: >30% extra sample efficiency, <3% wallclock overheadβ210Updated last week
- Normalized Transformer (nGPT)β145Updated last month
- WIPβ92Updated 5 months ago
- β201Updated 6 months ago
- Implementation of Diffusion Transformer (DiT) in JAXβ261Updated 7 months ago
- Accelerated First Order Parallel Associative Scanβ169Updated 4 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation preconditionβ¦β146Updated last month
- For optimization algorithm research and development.β484Updated this week
- Annotated version of the Mamba paperβ469Updated 10 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.β90Updated last month
- β75Updated 6 months ago
- Ο-GPT: A New Approach to Autoregressive Modelsβ61Updated 5 months ago
- Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAIβ270Updated 2 months ago
- A library for unit scaling in PyTorchβ118Updated last month
- β149Updated 5 months ago
- β25Updated last year
- Universal Tensor Operations in Einstein-Inspired Notation for Python.β341Updated last month
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT trainingβ121Updated 9 months ago
- DeMo: Decoupled Momentum Optimizationβ170Updated last month
- Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adamβ73Updated 5 months ago
- Experiment of using Tangent to autodiff tritonβ74Updated 11 months ago
- Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)β182Updated 7 months ago
- $100K or 100 Days: Trade-offs when Pre-Training with Academic Resourcesβ119Updated this week
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvementβ¦β345Updated this week
- β296Updated 6 months ago