google/brax

Что это?

Ключевые возможности

• 01Запускает симуляции твёрдых тел на JAX с использованием GPU и TPU и масштабирует их на несколько устройств.
• 02Предоставляет четыре взаимозаменяемых физических конвейера (MJX, обобщённый, позиционный и пружинный) с единым API.
• 03Включает PPO, SAC, ARS, эволюционные стратегии и аналитические градиенты политики для обучения.
• 04Поддерживает дифференцируемые симуляции, позволяя градиентам проходить через шаги физического моделирования.
• 05Предлагает ноутбуки Colab с основами, обучением, MJX и интеграцией с PyTorch.

Сценарии использования

• Обучение политик передвижения или манипуляции для симулируемых роботов с помощью обучения с подкреплением.
• Запуск больших пакетов параллельных физических симуляций на аппаратных ускорителях.
• Эксперименты с дифференцируемой физикой и аналитическими методами градиента политики.
• Использование симулятора в качестве среды обучения внутри PyTorch или других ML-фреймворков.

Быстрый старт

1. 1

   Создать и активировать виртуальное окружение

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

   Обновить pip

   pip install --upgrade pip
3. 3

   Установить Brax из PyPI

   pip install brax
4. 4

   Обучить модель

   learn

Что важно знать

• !Активно поддерживается только brax/training; brax/envs заменён на MuJoCo Playground.
• !Для обучения на GPU NVIDIA необходимо отдельно установить CUDA, cuDNN и JAX с поддержкой GPU.

В подборках

Темы

Частые вопросы

Что такое brax?

Физический движок на базе JAX для массово-параллельного дифференцируемого моделирования твёрдых тел на ускорителях в исследованиях робототехники и RL. Brax — физический движок на JAX, рассчитанный на работу на ускорителях, таких как GPU и TPU. Он предназначен для исследований в робототехнике, обучении с подкреплением, восприятии человеком и материаловедении и позволяет распределять симуляции между несколькими устройствами.

brax можно использовать бесплатно? Какая у него лицензия?

brax распространяется с открытым исходным кодом под лицензией Apache-2.0. Перед использованием в коммерческих продуктах проверьте условия лицензии.

brax всё ещё поддерживается?

Активно поддерживается. Последний коммит в brax — 20 окт. 2026 г..

Для чего можно использовать brax?

Обучение политик передвижения или манипуляции для симулируемых роботов с помощью обучения с подкреплением. Запуск больших пакетов параллельных физических симуляций на аппаратных ускорителях. Эксперименты с дифференцируемой физикой и аналитическими методами градиента политики. Использование симулятора в качестве среды обучения внутри PyTorch или других ML-фреймворков.

Как установить brax?

Создать и активировать виртуальное окружение: python3 -m venv env && source env/bin/activate Обновить pip: pip install --upgrade pip Установить Brax из PyPI: pip install brax Обучить модель: learn

Где скачать brax?

Исходный код и инструкции по установке — на GitHub: https://github.com/google/brax