hiverge / cifar10-speedrunLinks
CIFAR-10 speedrun: Trains to 94% accuracy in 1.98 seconds on a single NVIDIA A100 GPU.
☆55Updated 3 months ago
Alternatives and similar repositories for cifar10-speedrun
Users that are interested in cifar10-speedrun are comparing it to the libraries listed below
Sorting:
- Landing repository for the paper "Softpick: No Attention Sink, No Massive Activations with Rectified Softmax"☆86Updated 4 months ago
- Extending the Context of Pretrained LLMs by Dropping Their Positional Embedding☆193Updated 2 weeks ago
- ☆109Updated 6 months ago
- An implementation of PSGD Kron second-order optimizer for PyTorch☆98Updated 6 months ago
- Official repo for BWLer: Barycentric Weight Layer☆29Updated 4 months ago
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆110Updated 10 months ago
- train with kittens!☆63Updated last year
- H-Net Dynamic Hierarchical Architecture☆81Updated 4 months ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆132Updated last year
- Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)☆198Updated last year
- Jax like function transformation engine but micro, microjax☆34Updated last year
- WIP☆93Updated last year
- The Automated LLM Speedrunning Benchmark measures how well LLM agents can reproduce previous innovations and discover new ones in languag…☆127Updated 3 months ago
- Training code for Sparse Autoencoders on Embedding models☆39Updated 11 months ago
- Universal Reasoning Model☆121Updated 2 weeks ago
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆103Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆92Updated 2 years ago
- ☆70Updated last year
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆186Updated last week
- ☆92Updated last year
- $100K or 100 Days: Trade-offs when Pre-Training with Academic Resources☆150Updated 4 months ago
- Explorations into the recently proposed Taylor Series Linear Attention☆100Updated last year
- A collection of various llm pruning implementations, training code for GPUs & TPUs, and evaluation script.☆58Updated last month
- Supporting code for the blog post on modular manifolds.☆115Updated 4 months ago
- ☆123Updated 7 months ago
- ☆38Updated last year
- Attention Kernels for Symmetric Power Transformers☆129Updated 4 months ago
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆18Updated 6 months ago
- DeMo: Decoupled Momentum Optimization☆198Updated last year
- 📄Small Batch Size Training for Language Models☆80Updated 3 months ago