Erik Frey 390bce2352 Avoid placing arrays on device for unused MJX fields. Add `device` parameter to mjx.make_data. ...

PiperOrigin-RevId: 663808411
Change-Id: Ic55875f87f5c36b9dfe21e5ed5891d64c7003cab

2024-08-16 12:02:21 -07:00

..

__init__.py

Add MJX and bump version number to 3.0.0.

2023-10-18 10:33:29 +01:00

collision_convex.py

Speed up hfield JIT time by reducing XLA code generation overhead.

2024-07-18 15:01:43 -07:00

collision_driver_test.py

Add hfield. Fixes #1655 #1491 #1695

2024-05-29 16:21:34 -07:00

collision_driver.py

Add hfield. Fixes #1655 #1491 #1695

2024-05-29 16:21:34 -07:00

collision_primitive.py

Prepare MJX for condim.

2024-04-18 12:43:36 -07:00

collision_sdf.py

Speed up primitive sdf slightly.

2024-05-24 20:01:54 -07:00

collision_types.py

Add hfield. Fixes #1655 #1491 #1695

2024-05-29 16:21:34 -07:00

constraint_test.py

Add efc_pos to MJX.

2024-08-15 10:33:44 -07:00

constraint.py

Add efc_pos to MJX.

2024-08-15 10:33:44 -07:00

dataclasses.py

Add hfield. Fixes #1655 #1491 #1695

2024-05-29 16:21:34 -07:00

forward_test.py

Fixes to the MJX Euler integrator and Newton solver, and test improvements to catch future bugs like this.

2024-02-12 15:46:27 -08:00

forward.py

Add sensors to MJX. Implements jointpos and actuatorpos as examples.

2024-08-08 10:33:19 -07:00

io_test.py

Adds support for fixed tendons to MJX.

2024-07-22 10:49:13 -07:00

io.py

Avoid placing arrays on device for unused MJX fields. Add `device` parameter to mjx.make_data.

2024-08-16 12:02:21 -07:00

math_test.py

Add MJX and bump version number to 3.0.0.

2023-10-18 10:33:29 +01:00

math.py

Add ellipsoid sdf collisions to MJX.

2024-04-10 01:41:10 -07:00

mesh_test.py

Prepare MJX for condim.

2024-04-18 12:43:36 -07:00

mesh.py

Add hfield. Fixes #1655 #1491 #1695

2024-05-29 16:21:34 -07:00

passive_test.py

Adds support for fixed tendons to MJX.

2024-07-22 10:49:13 -07:00

passive.py

Adds support for fixed tendons to MJX.

2024-07-22 10:49:13 -07:00

ray_test.py

Fix bug in `mjx.ray` and speed up ray-mesh by ~30% for large meshes.

2024-04-29 15:48:00 -07:00

ray.py

Fix bug in `mjx.ray` and speed up ray-mesh by ~30% for large meshes.

2024-04-29 15:48:00 -07:00

scan_test.py

Remove deprecated mjx.device_* functions. They are not suited to upcoming condim support.

2024-04-19 15:34:28 -07:00

scan.py

Fix jax deprecation warning for jax.tree_map.

2024-04-18 11:13:35 -07:00

sensor_test.py

Modify MJX sensor_pos to handle unsupported sensors.

2024-08-16 10:10:37 -07:00

sensor.py

Modify MJX sensor_pos to handle unsupported sensors.

2024-08-16 10:10:37 -07:00

smooth_test.py

Adds support for fixed tendons to MJX.

2024-07-22 10:49:13 -07:00

smooth.py

Adds support for fixed tendons to MJX.

2024-07-22 10:49:13 -07:00

solver_test.py

Elliptic friction in MJX.

2024-06-07 16:05:58 -07:00

solver.py

Elliptic friction in MJX.

2024-06-07 16:05:58 -07:00

support_test.py

Add contact force support function. #1555

2024-05-10 13:41:45 -07:00

support.py

Add hfield. Fixes #1655 #1491 #1695

2024-05-29 16:21:34 -07:00

test_util.py

Elliptic friction in MJX.

2024-06-07 16:05:58 -07:00

types.py

Avoid placing arrays on device for unused MJX fields. Add `device` parameter to mjx.make_data.

2024-08-16 12:02:21 -07:00