Skip to main content

MuJoCo XLA (MJX)

Project description

MuJoCo XLA (MJX)

PyPI Python Version PyPI version

This package is a re-implementation of the MuJoCo physics engine in JAX. This library is developed and maintained by Google DeepMind, and is kept up-to-date with the latest developments in MuJoCo itself.

The mujoco-mjx package is API-compatible with MuJoCo, but is missing some features found in MuJoCo. See our documentation for more details concerning feature parity.

Installation

The recommended way to install this package is via PyPI:

pip install mujoco-mjx

Usage

Once installed, the package can be imported via from mujoco import mjx. Please consult our documentation for further detail on the package's API.

We recommend going through the tutorial notebook which introduces the MJX API and trains a reinforcement learning policy in a few minutes: Open In Colab

Versioning

The major.minor.micro portion of the version number matches the version of MuJoCo that this library provides. Optionally, if we release updates to MJX that target the same version of MuJoCo, a .postN suffix is added, for example 3.0.1.post2 represents the second update to MJX for MuJoCo 3.0.1.

License and Disclaimer

Copyright 2023 DeepMind Technologies Limited

MuJoCo and its libraries are licensed under the Apache License, Version 2.0. You may obtain a copy of the License at https://www.apache.org/licenses/LICENSE-2.0.

This is not an officially supported Google product.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

mujoco-mjx-3.1.3.tar.gz (6.4 MB view hashes)

Uploaded Source

Built Distribution

mujoco_mjx-3.1.3-py3-none-any.whl (6.4 MB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page