Repolode
投稿する
google-research

google-research/kauldron

研究開発の高速な反復を支える、JAX/Flaxでの機械学習モデル学習向けモジュール型・スケーラブルなライブラリ。

GitHub で見る公式サイト
スター
359
フォーク
36
未解決の Issue
83
最終コミット
昨日

これは何？

Kauldronは、研究開発の速度とモジュール性を重視して設計された、機械学習モデル学習用のPythonライブラリです。各コンポーネントは独立しており、フレームワークの外でも利用できるため、独自のデータパイプラインやFlaxモデル、オプティマイザーと組み合わせられます。

TFDS、Grain、SeqIOなどのデータセット、任意のFlaxモデル、任意のオプティマイザーをつなぐ役割を果たします。設定はカスタマイズしたり上書きしたりでき、たとえばモデルアーキテクチャのスイープや、内部レイヤーのパラメーター変更が可能です。MNISTオートエンコーダーなどの設定例を出発点として利用できます。

素早いプロトタイピングを可能にする、Colabを中心としたワークフロー向けに作られています。組み込み機能には、プロット連携、プロファイラー、事後デバッグ、実行時のshapeチェックがあります。

主な機能

  • 01TFDS、Grain、SeqIO、または独自のデータパイプラインを利用できます。
  • 02任意のFlaxモデルとオプティマイザーに対応します。
  • 03設定はスイープでき、あらゆる階層で上書きできます。
  • 04コンポーネントはフレームワークの外でも個別に利用できます。
  • 05プロファイラーと実行時のshapeチェックを備えています。

ユースケース

  • Colabノートブックで新しいモデルアーキテクチャをプロトタイピングする。
  • 機械学習研究でハイパーパラメーターやアーキテクチャのスイープを実行する。
  • 標準化された環境で、独自データセットを使ってFlaxモデルを学習する。

使う前に知っておきたいこと

  • READMEには、Googleが公式にサポートする製品ではないと記載されています。
  • READMEには、インストール方法や実行方法の説明がありません。

トピック

#deep-learning#jax#flax#machine-learning#training-framework#python

よくある質問

kauldron とは？
研究開発の高速な反復を支える、JAX/Flaxでの機械学習モデル学習向けモジュール型・スケーラブルなライブラリ。 Kauldronは、研究開発の速度とモジュール性を重視して設計された、機械学習モデル学習用のPythonライブラリです。各コンポーネントは独立しており、フレームワークの外でも利用できるため、独自のデータパイプラインやFlaxモデル、オプティマイザーと組み合わせられます。
kauldron は無料で使えますか？ライセンスは？
kauldron は Apache-2.0 ライセンスのオープンソースです。商用製品で使う前にライセンス条件を確認してください。
kauldron は今もメンテナンスされていますか？
活発にメンテナンス中。kauldron の最終コミットは 2026年11月6日 です。
kauldron は何に使えますか？
Colabノートブックで新しいモデルアーキテクチャをプロトタイピングする。 機械学習研究でハイパーパラメーターやアーキテクチャのスイープを実行する。 標準化された環境で、独自データセットを使ってFlaxモデルを学習する。
kauldron はどこで入手できますか？
ソースコードとインストール手順は GitHub にあります：https://github.com/google-research/kauldron

概要はプロジェクトの README をもとに AI が生成しています。GitHub の数値は毎日更新されます。

関連プロジェクト