Google, 연구 속도에 최적화된 ML 학습 라이브러리 Kauldron 공개

Kauldron: Modular, scalable library to train ML models

Google Research가 공개한 Kauldron은 머신러닝 모델 학습을 위한 라이브러리로, 연구 속도와 모듈성을 핵심으로 내세운다. 모든 구성 요소가 독립적으로 동작해 TFDS, Grain, SeqIO 등 어떤 데이터셋과 flax 모델, 옵티마이저든 자유롭게 조합할 수 있고, 내부 레이어 파라미터까지 덮어쓸 수 있다. 예제 설정을 바로 실행하는 Colab 우선 워크플로와 XM 플롯, 프로파일러, 런타임 형상 검사 등을 제공한다. 다만 공식 지원 Google 제품은 아니라고 명시했다.