pyrox.dev
1{
2 lib,
3 buildPythonPackage,
4
5 # src / metadata
6 mujoco-main,
7
8 # build-system
9 setuptools,
10
11 # dependencies
12 absl-py,
13 etils,
14 importlib-resources,
15 jax,
16 jaxlib,
17 mujoco,
18 scipy,
19 trimesh,
20}:
21
22buildPythonPackage {
23 pname = "mujoco-mjx";
24 inherit (mujoco-main) src version;
25
26 pyproject = true;
27
28 sourceRoot = "${mujoco-main.src.name}/mjx";
29
30 build-system = [ setuptools ];
31
32 dependencies = [
33 absl-py
34 etils
35 importlib-resources
36 jax
37 jaxlib
38 mujoco
39 scipy
40 trimesh
41 ]
42 ++ etils.optional-dependencies.epath;
43
44 pythonImportsCheck = [ "mujoco.mjx" ];
45
46 meta = {
47 description = "MuJoCo XLA (MJX)";
48 inherit (mujoco.meta) homepage changelog license;
49 maintainers = with lib.maintainers; [ nim65s ];
50 };
51}