Фреймворк для JIT-компиляции обычных Python-функций в ядра, работающие на CPU или GPU NVIDIA, для симуляций, робототехники и машинного обучения.
google-deepmind/mujoco_warp
Версия физического симулятора MuJoCo с ускорением на GPU, созданная на базе NVIDIA Warp для высокопроизводительного моделирования робототехники.
- Звёзды
- 1,607
- Форки
- 271
- Открытые issues
- 104
- Последний коммит
- 7 часов назад
Что это?
MuJoCo Warp (MJWarp) — порт физического симулятора MuJoCo с ускорением на GPU, предназначенный для оборудования NVIDIA. Он рассчитан на исследовательские задачи в робототехнике, где требуется параллельно выполнять множество шагов симуляции; для разработки и отладки его также можно запускать на CPU.
Проект поддерживает твёрдые тела с контактами, а также мягкие тела, ткань и поля знаковых расстояний. Его совместно поддерживают Google DeepMind и NVIDIA в рамках проекта физического движка Newton. Во многих существующих конфигурациях MuJoCo его можно использовать как drop-in replacement; установить пакет можно из PyPI под именем mujoco-warp.
В MJWarp также есть пакетный рендерер с трассировкой лучей для камер во множестве параллельных миров. Он поддерживает меши, текстуры, поля высот, Gaussian splats и тени. Некоторые возможности MuJoCo пока не поддерживаются: решатель PGS, приводы и датчики типа PLUGIN, а также дифференцируемость через Warp.
Ключевые возможности
- 01Запускает совместимую с MuJoCo физическую симуляцию на GPU NVIDIA; для разработки доступен запуск на CPU.
- 02Моделирует твёрдые тела с контактами, а также мягкие тела, ткань и поля знаковых расстояний.
- 03Включает пакетный рендерер с трассировкой лучей для множества параллельных камер, поддерживающий меши, текстуры и тени.
- 04Устанавливается из PyPI и во многих проектах может заменить MuJoCo без изменений в интеграции.
- 05Предоставляет CLI-инструмент для бенчмарков с возможностью создавать трассировки событий для профилирования производительности.
Сценарии использования
- Запуск множества параллельных симуляций роботов для обучения с подкреплением на GPU NVIDIA.
- Рендеринг изображений с камер во множестве симулируемых сред для обучения роботов на основе компьютерного зрения.
- Ускорение существующих исследовательских конвейеров в робототехнике на базе MuJoCo с минимальными изменениями кода.
Быстрый старт
- 1
Клонируйте репозиторий
git clone https://github.com/google-deepmind/mujoco_warp.git && cd mujoco_warp - 2
Запустите просмотрщик гуманоидных роботов локально
python benchmarks/run.py -f unitree_g1_flat --view - 3
Или установите пакет из PyPI
pip install mujoco-warp
Что важно знать
- Для быстрой симуляции требуется GPU NVIDIA; CPU поддерживается только для разработки и отладки.
- Некоторые функции MuJoCo не поддерживаются, включая интегратор средней точки IMPLICITFAST, решатель PGS, а также приводы и датчики PLUGIN.
- Дифференцируемость с помощью Warp пока недоступна.
В подборках
Темы
Частые вопросы
- Что такое mujoco_warp?
- Версия физического симулятора MuJoCo с ускорением на GPU, созданная на базе NVIDIA Warp для высокопроизводительного моделирования робототехники. MuJoCo Warp (MJWarp) — порт физического симулятора MuJoCo с ускорением на GPU, предназначенный для оборудования NVIDIA. Он рассчитан на исследовательские задачи в робототехнике, где требуется параллельно выполнять множество шагов симуляции; для разработки и отладки его также можно запускать на CPU.
- mujoco_warp можно использовать бесплатно? Какая у него лицензия?
- mujoco_warp распространяется с открытым исходным кодом под лицензией Apache-2.0. Перед использованием в коммерческих продуктах проверьте условия лицензии.
- mujoco_warp всё ещё поддерживается?
- Активно поддерживается. Последний коммит в mujoco_warp — 6 нояб. 2026 г..
- Для чего можно использовать mujoco_warp?
- Запуск множества параллельных симуляций роботов для обучения с подкреплением на GPU NVIDIA. Рендеринг изображений с камер во множестве симулируемых сред для обучения роботов на основе компьютерного зрения. Ускорение существующих исследовательских конвейеров в робототехнике на базе MuJoCo с минимальными изменениями кода.
- Как установить mujoco_warp?
- Клонируйте репозиторий: git clone https://github.com/google-deepmind/mujoco_warp.git && cd mujoco_warp Запустите просмотрщик гуманоидных роботов локально: python benchmarks/run.py -f unitree_g1_flat --view Или установите пакет из PyPI: pip install mujoco-warp
- Где скачать mujoco_warp?
- Исходный код и инструкции по установке — на GitHub: https://github.com/google-deepmind/mujoco_warp
Описание создано ИИ на основе README проекта. Статистика GitHub обновляется ежедневно.
Похожие проекты
Физический движок для GPU-симуляции на базе NVIDIA Warp, предназначенный для специалистов по робототехнике и исследователей симуляций.
API для обучения роботов с подкреплением в стиле Isaac Lab, работающий на MuJoCo Warp с GPU-ускорением.
Универсальный физический движок для быстрой и точной симуляции контактирующих сочленённых тел, широко используемый в робототехнике и исследованиях машинного обучения.
ПО с открытым исходным кодом на Rust для управления маленьким двуногим роботом-уткой: движение, геймпад, обновления и видеопоток с камеры.
Физический движок на базе JAX для массово-параллельного дифференцируемого моделирования твёрдых тел на ускорителях в исследованиях робототехники и RL.