NVIDIA-NeMo / Emerging-OptimizersLinks
☆132Updated last 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☆229Updated 7 months ago
- supporting pytorch FSDP for optimizers☆84Updated last year
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆186Updated last week
- ☆124Updated last year
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆246Updated 7 months ago
- A library for unit scaling in PyTorch☆133Updated 6 months ago
- Fast and memory-efficient exact attention☆75Updated 10 months ago
- Accelerated First Order Parallel Associative Scan☆196Updated 3 weeks ago
- Muon fsdp 2☆51Updated 5 months ago
- ☆92Updated last year
- ☆83Updated 2 years ago
- Supporting code for the blog post on modular manifolds.☆113Updated 4 months ago
- Experiment of using Tangent to autodiff triton☆82Updated 2 years ago
- The evaluation framework for training-free sparse attention in LLMs☆110Updated 3 months ago
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆131Updated last year
- ☆133Updated 8 months ago
- Normalized Transformer (nGPT)☆197Updated last year
- Triton-based implementation of Sparse Mixture of Experts.☆262Updated 3 months ago
- ☆104Updated 11 months ago
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆88Updated last year
- A MAD laboratory to improve AI architecture designs 🧪☆135Updated last year
- Understand and test language model architectures on synthetic tasks.☆251Updated 2 weeks ago
- ☆52Updated last month
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆75Updated last year
- This repository contains the experimental PyTorch native float8 training UX☆227Updated last year
- ☆269Updated 7 months ago
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆128Updated 7 months ago
- ring-attention experiments☆163Updated last year
- JAX bindings for Flash Attention v2☆103Updated last month
- ☆44Updated 2 months ago