google/brax

これは何？

主な機能

• 01JAXを使ってGPUやTPU上で剛体シミュレーションを実行し、複数デバイスにスケールできます。
• 021つのAPIから切り替えて使える4種類の物理パイプライン（MJX、generalized、positional、spring）を備えています。
• 03学習用にPPO、SAC、ARS、進化戦略、解析的方策勾配が含まれています。
• 04微分可能なシミュレーションに対応し、物理ステップを通じて勾配を伝播できます。
• 05基礎、学習、MJX、PyTorch連携を扱うColabノートブックを利用できます。

ユースケース

• シミュレーション上のロボットに対し、強化学習で歩行やマニピュレーションの方策を学習する。
• アクセラレータ上で物理シミュレーションを大規模に並列実行する。
• 微分可能な物理シミュレーションや解析的方策勾配の手法を試す。
• PyTorchなどのMLフレームワーク内で、学習環境としてシミュレータを使う。

クイックスタート

1. 1

   仮想環境を作成して有効化する

   python3 -m venv env && source env/bin/activate
2. 2

   pipをアップグレードする

   pip install --upgrade pip
3. 3

   PyPIからBraxをインストールする

   pip install brax
4. 4

   モデルを学習させる

   learn

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

• !現在も積極的にメンテナンスされているのはbrax/trainingのみです。brax/envsはMuJoCo Playgroundに置き換えられています。
• !NVIDIA GPUで学習するには、CUDA、cuDNN、GPU対応版JAXを別途インストールする必要があります。

収録されている特集

トピック

よくある質問

brax とは？

JAXベースの剛体物理エンジン。アクセラレータ上で大規模並列かつ微分可能なシミュレーションを実行でき、ロボティクスやRLの研究に使えます。 BraxはJAXで実装された、GPUやTPUなどのアクセラレータ向け物理エンジンです。ロボティクス、強化学習、人間の知覚、材料科学の研究を対象としており、複数デバイスにまたがるシミュレーションにも対応します。

brax は無料で使えますか？ライセンスは？

brax は Apache-2.0 ライセンスのオープンソースです。商用製品で使う前にライセンス条件を確認してください。

brax は今もメンテナンスされていますか？

活発にメンテナンス中。brax の最終コミットは 2026年10月20日 です。

brax は何に使えますか？

シミュレーション上のロボットに対し、強化学習で歩行やマニピュレーションの方策を学習する。 アクセラレータ上で物理シミュレーションを大規模に並列実行する。 微分可能な物理シミュレーションや解析的方策勾配の手法を試す。 PyTorchなどのMLフレームワーク内で、学習環境としてシミュレータを使う。

brax のインストール方法は？

仮想環境を作成して有効化する: python3 -m venv env && source env/bin/activate pipをアップグレードする: pip install --upgrade pip PyPIからBraxをインストールする: pip install brax モデルを学習させる: learn

brax はどこで入手できますか？

ソースコードとインストール手順は GitHub にあります：https://github.com/google/brax