Google выпустила Kauldron — фреймворк для обучения ML-моделей, заточенный под скорость исследований
Kauldron: Modular, scalable library to train ML models
Kauldron — открытая библиотека от google-research для обучения моделей на JAX. Каждый её компонент самодостаточен и работает вне фреймворка: подключайте любой датасет (TFDS, Grain, SeqIO или собственный пайплайн), любую модель на flax и любой оптимизатор. Всё можно переопределять — вплоть до параметров отдельных слоёв. Примеры конфигов запускаются из коробки, а Colab-ориентированный workflow, встроенные XM-графики и профайлер ускоряют прототипирование.
Kauldron — это библиотека для обучения моделей машинного обучения, оптимизированная под скорость исследований и модульность.