☆34Sep 10, 2024Updated last year
Alternatives and similar repositories for zeroshampoo
Users that are interested in zeroshampoo are comparing it to the libraries listed below
Sorting:
- Load & manage evolving datasets efficiently☆23Aug 22, 2025Updated 6 months ago
- supporting pytorch FSDP for optimizers☆84Dec 8, 2024Updated last year
- research impl of Native Sparse Attention (2502.11089)☆63Feb 19, 2025Updated last year
- ☆93Jul 5, 2024Updated last year
- ☆13Jun 3, 2024Updated last year
- WIP☆94Aug 13, 2024Updated last year
- ☆23Jun 18, 2024Updated last year
- ☆33Nov 4, 2024Updated last year
- Optimized primitives for collective multi-GPU communication☆10May 8, 2024Updated last year
- ☆11Feb 26, 2024Updated 2 years ago
- ☆53Jan 6, 2024Updated 2 years ago
- ☆48Jan 18, 2024Updated 2 years ago
- ☆24Jun 4, 2024Updated last year
- coloring terminal text with intensities (used for plotting probability, entropy with tokens)☆12Oct 11, 2024Updated last year
- Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam☆86Jul 28, 2024Updated last year
- The 2D discrete wavelet transform for JAX☆45Feb 28, 2023Updated 3 years ago
- An AI character interaction system with emotional modeling and advanced memory management☆17Oct 26, 2024Updated last year
- Focused on fast experimentation and simplicity☆80Dec 24, 2024Updated last year
- ☆13Sep 13, 2023Updated 2 years ago
- Code for the paper "Function-Space Learning Rates"☆25Jun 3, 2025Updated 9 months ago
- Efficient optimizers☆294Updated this week
- ☆68Mar 21, 2025Updated 11 months ago
- train with kittens!☆64Oct 25, 2024Updated last year
- ☆17Apr 17, 2024Updated last year
- Train VAE like a boss☆313Oct 21, 2024Updated last year
- ☆14Mar 31, 2025Updated 11 months ago
- ☆21Feb 24, 2026Updated 3 weeks ago
- ☆23Oct 15, 2024Updated last year
- Highly specialized crate to parse and use `google/sentencepiece` 's precompiled_charsmap in `tokenizers`☆21Jan 8, 2026Updated 2 months ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆132Apr 17, 2024Updated last year
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆92Oct 30, 2024Updated last year
- a jax benchmark for ad hoc teamwork☆20Updated this week
- Dion optimizer algorithm☆453Jan 16, 2026Updated 2 months ago
- Official repo of dataset-decomposition paper [NeurIPS 2024]☆21Jan 8, 2025Updated last year
- generative models on toys☆12Sep 10, 2024Updated last year
- [NeurIPS 2022] Supervising the Multi-Fidelity Race of Hyperparameter Configurations☆13Apr 25, 2023Updated 2 years ago
- Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun☆57Mar 10, 2025Updated last year
- HGRN2: Gated Linear RNNs with State Expansion☆56Aug 20, 2024Updated last year
- ☆30Dec 2, 2024Updated last year