Modular, scalable library to train ML models
☆213Feb 25, 2026Updated this week
Alternatives and similar repositories for kauldron
Users that are interested in kauldron are comparing it to the libraries listed below
Sorting:
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆549Updated this week
- Minimal yet performant LLM examples in pure JAX☆240Jan 14, 2026Updated last month
- Two implementations of ZeRO-1 optimizer sharding in JAX☆14Jun 11, 2023Updated 2 years ago
- ☆314Updated this week
- CLU lets you write beautiful training loops in JAX.☆367Updated this week
- Pytorch-like dataloaders for JAX.☆101Dec 16, 2025Updated 2 months ago
- ☆77Feb 23, 2026Updated last week
- Accelerate, Optimize performance with streamlined training and serving options with JAX.☆339Updated this week
- ☆270Feb 23, 2026Updated last week
- MiniMax-Provider-Verifier offers a rigorous, vendor-agnostic way to verify whether third-party deployments of the Minimax M2 model are co…☆28Feb 18, 2026Updated last week
- JAX-Toolbox☆385Feb 24, 2026Updated last week
- A simple, performant and scalable Jax LLM!☆2,148Feb 24, 2026Updated last week
- A JAX library for building lattice-based speech transducer models☆46Updated this week
- Orbax provides common checkpointing and persistence utilities for JAX users☆483Updated this week
- jax-triton contains integrations between JAX and OpenAI Triton☆439Feb 24, 2026Updated last week
- ☆20Sep 5, 2025Updated 5 months ago
- ☆24Dec 16, 2024Updated last year
- ☆164Dec 13, 2023Updated 2 years ago
- ☆15May 11, 2025Updated 9 months ago
- a viewer for for lancedb. including some actions like CRUD etc☆12Apr 27, 2025Updated 10 months ago
- Intuitive graphical representation of source code☆14Mar 15, 2023Updated 2 years ago
- Basic library for spatial audio SOFA files☆12Sep 29, 2020Updated 5 years ago
- High-performance tokenized language data-loader for Python C++ extension☆14Jul 22, 2024Updated last year
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆695Jan 26, 2026Updated last month
- a Jax quantization library☆93Feb 23, 2026Updated last week
- An interactive HTML pretty-printer for machine learning research in IPython notebooks.☆460Aug 8, 2025Updated 6 months ago
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,869Jun 22, 2025Updated 8 months ago
- A text embedding extension for the Polars Dataframe library.☆29Nov 21, 2024Updated last year
- When Visualizing is the First Step to Reasoning: MIRA, a Benchmark for Visual Chain-of-Thought☆27Feb 14, 2026Updated 2 weeks ago
- Code for the paper Joint Discovery of Object States and Manipulation Actions, ICCV 2017☆14Aug 7, 2018Updated 7 years ago
- A python package to efficiently extract linguistic features for text/NLP datasets☆24Feb 18, 2026Updated last week
- DiT (training + flow matching) in Jax☆11Jan 5, 2025Updated last year
- ☆21Feb 3, 2026Updated last month
- PIX is an image processing library in JAX, for JAX.☆434Mar 6, 2025Updated 11 months ago
- HomebrewNLP in JAX flavour for maintable TPU-Training☆51Jan 20, 2024Updated 2 years ago
- ☆348Feb 19, 2026Updated last week
- This is a new metric that can be used to evaluate faithfulness of text generated by LLMs. The work behind this repository can be found he…☆31Aug 25, 2023Updated 2 years ago
- A platform for managing machine learning experiments☆898Feb 6, 2026Updated 3 weeks ago
- Gemma open-weight LLM library, from Google DeepMind☆4,053Feb 23, 2026Updated last week