evanatyourservice / kron_torchLinks
An implementation of PSGD Kron second-order optimizer for PyTorch
☆97Updated 3 months ago
Alternatives and similar repositories for kron_torch
Users that are interested in kron_torch are comparing it to the libraries listed below
Sorting:
- Getting crystal-like representations with harmonic loss☆192Updated 7 months ago
- supporting pytorch FSDP for optimizers☆83Updated 11 months ago
- Efficient optimizers☆275Updated last week
- ☆222Updated 11 months ago
- 🧱 Modula software package☆303Updated 3 months ago
- ☆68Updated last year
- WIP☆93Updated last year
- ☆81Updated last year
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆103Updated 10 months ago
- The AdEMAMix Optimizer: Better, Faster, Older.☆186Updated last year
- Official implementation of the paper: "ZClip: Adaptive Spike Mitigation for LLM Pre-Training".☆139Updated last month
- DeMo: Decoupled Momentum Optimization☆197Updated 11 months ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆72Updated 6 months ago
- ☆119Updated 5 months ago
- ☆150Updated last year
- Scalable and Performant Data Loading☆335Updated last week
- FlashRNN - Fast RNN Kernels with I/O Awareness☆163Updated last month
- Focused on fast experimentation and simplicity☆75Updated 10 months ago
- ☆105Updated 3 months ago
- ☆91Updated last year
- ☆310Updated last year
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆132Updated last year
- 📄Small Batch Size Training for Language Models☆63Updated last month
- Landing repository for the paper "Softpick: No Attention Sink, No Massive Activations with Rectified Softmax"☆85Updated 2 months ago
- $100K or 100 Days: Trade-offs when Pre-Training with Academic Resources☆148Updated last month
- Dion optimizer algorithm☆384Updated this week
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆123Updated last year
- research impl of Native Sparse Attention (2502.11089)☆63Updated 9 months ago
- Supporting code for the blog post on modular manifolds.☆102Updated last month
- Fast, Modern, and Low Precision PyTorch Optimizers☆116Updated 2 months ago