evanatyourservice / kron_torchLinks
An implementation of PSGD Kron second-order optimizer for PyTorch
☆96Updated last month
Alternatives and similar repositories for kron_torch
Users that are interested in kron_torch are comparing it to the libraries listed below
Sorting:
- Efficient optimizers☆261Updated last month
- supporting pytorch FSDP for optimizers☆84Updated 9 months ago
- DeMo: Decoupled Momentum Optimization☆190Updated 9 months ago
- 🧱 Modula software package☆237Updated 3 weeks ago
- Getting crystal-like representations with harmonic loss☆194Updated 5 months ago
- The AdEMAMix Optimizer: Better, Faster, Older.☆186Updated last year
- ☆82Updated last year
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆101Updated 8 months ago
- ☆210Updated 9 months ago
- WIP☆94Updated last year
- $100K or 100 Days: Trade-offs when Pre-Training with Academic Resources☆146Updated 3 months ago
- Official implementation of the paper: "ZClip: Adaptive Spike Mitigation for LLM Pre-Training".☆132Updated last week
- Scalable and Performant Data Loading☆299Updated this week
- ☆65Updated 10 months ago
- ☆118Updated 3 months ago
- ☆150Updated last year
- Dion optimizer algorithm☆338Updated last week
- 📄Small Batch Size Training for Language Models☆60Updated 2 weeks ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆132Updated last year
- NanoGPT-speedrunning for the poor T4 enjoyers☆71Updated 4 months ago
- For optimization algorithm research and development.☆536Updated this week
- Landing repository for the paper "Softpick: No Attention Sink, No Massive Activations with Rectified Softmax"☆84Updated this week
- ☆87Updated last year
- Focused on fast experimentation and simplicity☆75Updated 8 months ago
- σ-GPT: A New Approach to Autoregressive Models☆67Updated last year
- Modular, scalable library to train ML models☆164Updated this week
- FlashRNN - Fast RNN Kernels with I/O Awareness☆97Updated 3 months ago
- ☆102Updated last month
- research impl of Native Sparse Attention (2502.11089)☆61Updated 6 months ago
- ☆53Updated last year