Google发布Kauldron：模块化ML训练库

Kauldron: Modular, scalable library to train ML models

Google Research开源了Kauldron，这是一个专为提升研究速度和模块化设计的机器学习模型训练库。它支持任意数据集、Flax模型和优化器，所有组件均可独立使用并灵活定制。Kauldron强调开箱即用的体验，提供Colab优先的工作流，集成XM图表、性能分析器和运行时形状检查等工具，让原型开发和迭代更加高效。无论是从MNIST自动编码器示例起步，还是自定义复杂架构，Kauldron都致力于成为连接各种组件的强力胶水，助力研究人员快速验证想法。