cloneofsimo / zeroshampooLinks
☆34Updated 9 months ago
Alternatives and similar repositories for zeroshampoo
Users that are interested in zeroshampoo are comparing it to the libraries listed below
Sorting:
- Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam☆80Updated 10 months ago
- ☆23Updated last year
- ☆51Updated last year
- Focused on fast experimentation and simplicity☆74Updated 6 months ago
- FID computation in Jax/Flax.☆27Updated 11 months ago
- The 2D discrete wavelet transform for JAX☆43Updated 2 years ago
- ☆19Updated last month
- supporting pytorch FSDP for optimizers☆82Updated 6 months ago
- Code for the paper "Function-Space Learning Rates"☆20Updated 3 weeks ago
- ☆78Updated 11 months ago
- Utilities for PyTorch distributed☆24Updated 3 months ago
- A JAX implementation of the continuous time formulation of Consistency Models☆85Updated 2 years ago
- Automatically take good care of your preemptible TPUs☆36Updated 2 years ago
- WIP☆93Updated 10 months ago
- ☆53Updated last year
- Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun☆52Updated 3 months ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆127Updated last year
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆44Updated 2 years ago
- Latent Diffusion Language Models☆68Updated last year
- ☆13Updated last year
- research impl of Native Sparse Attention (2502.11089)☆54Updated 4 months ago
- ☆21Updated 7 months ago
- ☆33Updated 7 months ago
- Exploring an idea where one forgets about efficiency and carries out attention across each edge of the nodes (tokens)☆51Updated 2 months ago
- ☆24Updated last month
- LayerNorm(SmallInit(Embedding)) in a Transformer to improve convergence☆60Updated 3 years ago
- Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto☆56Updated last year
- A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.☆18Updated 7 months ago
- JAX implementation of "Fine-Tuning Language Models with Just Forward Passes"☆19Updated 2 years ago
- ☆27Updated last year