Google、研究速度を最優先したMLトレーニングライブラリ「Kauldron」を公開

Kauldron: Modular, scalable library to train ML models

Google Researchが公開したKauldronは、JAXベースのMLモデル学習用ライブラリ。すべての部品が自己完結しており、TFDSやGrain、SeqIOなど任意のデータセット、任意のflaxモデル、任意のオプティマイザを組み合わせられる。モデル構造のスイープや内部レイヤーのパラメータ上書きも可能で、Colab中心のワークフローと充実したデバッグ機能で研究の反復速度を高める。