evanatyourservice / kron_torchLinks
An implementation of PSGD Kron second-order optimizer for PyTorch
β91Updated 2 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:
- supporting pytorch FSDP for optimizersβ82Updated 6 months ago
- π§± Modula software packageβ200Updated 2 months ago
- Efficient optimizersβ220Updated this week
- β188Updated 6 months ago
- β78Updated 11 months ago
- The AdEMAMix Optimizer: Better, Faster, Older.β183Updated 9 months ago
- β81Updated last year
- Getting crystal-like representations with harmonic lossβ190Updated 2 months ago
- WIPβ93Updated 10 months ago
- β104Updated last week
- Focused on fast experimentation and simplicityβ74Updated 6 months ago
- Fast, Modern, and Low Precision PyTorch Optimizersβ94Updated this week
- DeMo: Decoupled Momentum Optimizationβ188Updated 6 months ago
- β150Updated 10 months ago
- Official implementation of the paper: "ZClip: Adaptive Spike Mitigation for LLM Pre-Training".β127Updated 2 weeks ago
- β27Updated last year
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"β101Updated 6 months ago
- NanoGPT-speedrunning for the poor T4 enjoyersβ66Updated 2 months ago
- β98Updated 5 months ago
- Scalable and Performant Data Loadingβ277Updated this week
- β53Updated last year
- β61Updated 7 months ago
- FlashRNN - Fast RNN Kernels with I/O Awarenessβ91Updated last week
- The simplest, fastest repository for training/finetuning medium-sized GPTs.β134Updated this week
- $100K or 100 Days: Trade-offs when Pre-Training with Academic Resourcesβ140Updated last month
- A library for unit scaling in PyTorchβ125Updated 6 months ago
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of newβ¦β122Updated 10 months ago
- β55Updated 7 months ago
- β126Updated last month
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT trainingβ127Updated last year