NVIDIA-NeMo / Emerging-OptimizersLinks
☆114Updated 2 weeks ago
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☆225Updated 6 months ago
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆87Updated last year
- Muon fsdp 2☆47Updated 5 months ago
- The evaluation framework for training-free sparse attention in LLMs☆108Updated 2 months ago
- ☆124Updated last year
- JAX bindings for Flash Attention v2☆102Updated last week
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆181Updated 6 months ago
- supporting pytorch FSDP for optimizers☆84Updated last year
- A MAD laboratory to improve AI architecture designs 🧪☆135Updated last year
- Stick-breaking attention☆62Updated 6 months ago
- Understand and test language model architectures on synthetic tasks.☆248Updated 3 months ago
- 📄Small Batch Size Training for Language Models☆77Updated 3 months ago
- Accelerated First Order Parallel Associative Scan☆193Updated 2 weeks ago
- A library for unit scaling in PyTorch☆133Updated 5 months ago
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆74Updated last year
- ☆53Updated last year
- ☆83Updated 2 years ago
- ☆52Updated 3 weeks ago
- Fast and memory-efficient exact attention☆75Updated 10 months ago
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆131Updated last year
- ☆43Updated 2 months ago
- Normalized Transformer (nGPT)☆195Updated last year
- ☆102Updated 10 months ago
- nanoGPT-like codebase for LLM training☆113Updated 2 months ago
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆128Updated 6 months ago
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆244Updated 7 months ago
- Supporting code for the blog post on modular manifolds.☆109Updated 3 months ago
- ☆92Updated last year
- [ACL 2025] Outlier-Safe Pre-Training for Robust 4-Bit Quantization of Large Language Models☆34Updated 2 months ago
- Some preliminary explorations of Mamba's context scaling.☆218Updated last year