kauldron

Kauldron is a library for training machine learning models, optimized for research velocity and modularity.

Modularity:

All parts of Kauldron are self-contained, so can be used independently outside Kauldron.
Use any dataset (TFDS, Grain, SeqIO, your custom pipeline), any (flax) model, any optimizer,... Kauldron provides the glue that link everything together.
Everything can be customized and overwritten (e.g. sweep over models architecture, overwrite any inner layer parameter,...)

Research velocity:

Everything should work out-of the box. The example configs can be used and customized as a starting point.
Colab-first workflow for easy prototyping and fast iteration
Polished user experience (integrated XM plots, profiler, post-mortem debugging on borg, runtime shape checking, and many others...). Open an issue..

Citing Kauldron

If Kauldron was helpful for a publication, please cite this repository:

@software{kauldron2025github,
  author = {Klaus Greff and Etienne Pot and Mehdi S. M. Sajjadi},
  title = {{Kauldron}: A neural network training framework, optimized for research velocity and modularity.},
  url = {https://github.com/google-research/kauldron},
  version = {1.3.0},
  year = {2025},
}

This is not an officially supported Google product.

Metadata

Release files for kauldron 1.4.4

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for kauldron 1.4.4
File	Size	Uploaded
kauldron-1.4.4.tar.gz	396.1 kB	Jun 10, 2026	Details

Built distribution (wheel)

Table of built distributions (wheels) for kauldron 1.4.4
File	Interpreter	ABI	Platform	Reset
kauldron-1.4.4-py3-none-any.whl 609.3 kB Jun 10, 2026	Python 3	none	any	Details

Total release size: 1.0 MB

Release files / kauldron-1.4.4.tar.gz

Download URL	kauldron-1.4.4.tar.gz
Size	396.1 kB
Tags	Source
SHA-256 checksum How to use checksums	`4abf92cb7714a3f023849864c78f8a14f321a0638d2aef366b8a3ac694728e5b`
BLAKE2b-256 checksum How to use checksums	`2599455b3f8d810f9dd5f014c9d7a9d0df095ca48c379fa81c784416844ac349`
Upload date	Jun 10, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/6.2.0 CPython/3.12.13`

Release files / kauldron-1.4.4-py3-none-any.whl

Download URL	kauldron-1.4.4-py3-none-any.whl
Size	609.3 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`6d586b65de003369101c73bd55bbb64df38a396e511257be5a6e08658019fbc8`
BLAKE2b-256 checksum How to use checksums	`bf710d6aa91b77ca14a497cdb20890d2f008c8924346ce9ac734b20992ffdc72`
Upload date	Jun 10, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/6.2.0 CPython/3.12.13`