google-research / kauldronLinks
Modular, scalable library to train ML models
☆128Updated this week
Alternatives and similar repositories for kauldron
Users that are interested in kauldron are comparing it to the libraries listed below
Sorting:
- ☆270Updated 11 months ago
- A simple library for scaling up JAX programs☆139Updated 7 months ago
- ☆126Updated last month
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆139Updated this week
- An implementation of PSGD Kron second-order optimizer for PyTorch☆91Updated 2 months ago
- Library for reading and processing ML training data.☆463Updated this week
- Efficiently discovering algorithms via LLMs with evolutionary search and reinforcement learning.☆103Updated 2 months ago
- ☆53Updated last year
- ☆104Updated 2 weeks ago
- Implementation of Diffusion Transformer (DiT) in JAX☆278Updated last year
- ☆134Updated 2 months ago
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆34Updated 8 months ago
- ☆157Updated 2 months ago
- 🧱 Modula software package☆200Updated 3 months ago
- Focused on fast experimentation and simplicity☆75Updated 6 months ago
- Named Tensors for Legible Deep Learning in JAX☆181Updated last week
- $100K or 100 Days: Trade-offs when Pre-Training with Academic Resources☆140Updated last month
- ☆114Updated this week
- σ-GPT: A New Approach to Autoregressive Models