NVIDIA-NeMo / Emerging-OptimizersLinks
☆106Updated this week
Alternatives and similar repositories for Emerging-Optimizers
Users that are interested in Emerging-Optimizers are comparing it to the libraries listed below
Sorting:
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆223Updated 5 months ago
- Muon fsdp 2☆46Updated 4 months ago
- The evaluation framework for training-free sparse attention in LLMs☆106Updated 2 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆179Updated 5 months ago
- ☆121Updated last year
- Experiment of using Tangent to autodiff triton☆81Updated last year
- ☆100Updated 9 months ago
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆126Updated 5 months ago
- A MAD laboratory to improve AI architecture designs 🧪☆135Updated 11 months ago
- ☆49Updated last month
- supporting pytorch FSDP for optimizers☆84Updated last year
- JAX bindings for Flash Attention v2☆100Updated last month
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆86Updated last year
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆73Updated last year
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆91Updated 4 months ago
- Fast and memory-efficient exact attention☆74Updated 9 months ago
- A library for unit scaling in PyTorch☆132Updated 5 months ago
- ☆91Updated last year
- Understand and test language model architectures on synthetic tasks.☆245Updated 2 months ago
- 📄Small Batch Size Training for Language Models☆68Updated 2 months ago
- ☆53Updated last year
- Load compute kernels from the Hub☆348Updated this week
- Triton-based implementation of Sparse Mixture of Experts.☆253Updated 2 months ago
- ☆132Updated 6 months ago
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆243Updated 6 months ago
- Stick-breaking attention☆61Updated 5 months ago
- Normalized Transformer (nGPT)☆193Updated last year
- Accelerated First Order Parallel Associative Scan☆192Updated last year
- Code for studying the super weight in LLM☆121Updated last year
- Cold Compress is a hackable, lightweight, and open-source toolkit for creating and benchmarking cache compression methods built on top of…☆146Updated last year