# MuJoCo XLA (MJX)

[![PyPI Python Version][pypi-versions-badge]][pypi]
[![PyPI version][pypi-badge]][pypi]

[pypi-versions-badge]: https://img.shields.io/pypi/pyversions/mujoco-mjx
[pypi-badge]: https://badge.fury.io/py/mujoco-mjx.svg
[pypi]: https://pypi.org/project/mujoco-mjx/

本软件包是基于 [JAX](https://github.com/jax-ml/jax) 对 [MuJoCo 物理引擎](https://github.com/google-deepmind/mujoco) 的重新实现。该库由 Google DeepMind 开发和维护，并与 MuJoCo 本身的最新进展保持同步。

`mujoco-mjx` 软件包在 API 上与 MuJoCo 兼容，但缺少 MuJoCo 中的某些功能。有关功能特性的对齐情况，请参阅我们的[官方文档](https://mujoco.readthedocs.io/en/stable/mjx.html)。

## 安装

推荐通过 [PyPI](https://pypi.org/project/mujoco-mjx/) 安装本软件包：

```sh
pip install mujoco-mjx
```

## 使用说明

安装完成后，可以通过 `from mujoco import mjx` 导入该包。有关该包 API 的更多详细信息，请参阅我们的[官方文档](https://mujoco.readthedocs.io/en/stable/mjx.html)。

我们建议查阅教程 Notebook，它介绍了 MJX API 并在数分钟内训练一个强化学习策略：
[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google-deepmind/mujoco/blob/main/mjx/tutorial.ipynb)

## 版本规范

版本号的 `major.minor.micro` 部分与该库所基于的 MuJoCo 版本保持一致。可选地，如果我们发布了针对同一 MuJoCo 版本的 MJX 自身更新，则会添加 `.postN` 后缀，例如 `3.0.1.post2` 表示针对 MuJoCo 3.0.1 的 MJX 第二次更新。

## 开源许可证与免责声明

Copyright 2023 DeepMind Technologies Limited

MuJoCo 及其相关库基于 Apache 许可证 2.0 版（Apache License, Version 2.0）授权。您可在 https://www.apache.org/licenses/LICENSE-2.0 获取许可证副本。

本项目不是 Google 官方支持的产品。