Baruch Tabanpour a49f3f63fc Remove disable_jit in test. ...

PiperOrigin-RevId: 614824680
Change-Id: I15f1441dfb1a70b98e5a221c0e15b06572860748

2024-03-11 16:03:40 -07:00

..

__init__.py

Add MJX and bump version number to 3.0.0.

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

collision_base.py

Speed up convex-convex collisions.

2024-03-06 21:48:51 -08:00

collision_convex.py

Speed up convex-convex collisions.

2024-03-06 21:48:51 -08:00

collision_driver_test.py

Remove disable_jit in test.

2024-03-11 16:03:40 -07:00

collision_driver.py

Speed up convex-convex collisions.

2024-03-06 21:48:51 -08:00

collision_primitive.py

Add MJX and bump version number to 3.0.0.

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

constraint_test.py

Cleanup MJX tests and migrate them to put_model/put_data.

2023-12-05 23:08:05 -08:00

constraint.py

Adds support for mjOption.impratio to MJX.

2024-02-15 14:28:36 -08:00

dataclasses.py

Use less brittle search to determine whether an MJX field is a data or meta pytree node. Use correct type annotation for geom_convex_* fields in mjx.Model.

2024-02-14 13:57:55 -08:00

device_test.py

Adds support for explicit dense/sparse mass matrices to MJX.

2024-01-23 00:32:07 -08:00

device.py

Add cameras to MJX. Fixes #1422

2024-02-27 14:17:21 -08: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 cameras to MJX. Fixes #1422

2024-02-27 14:17:21 -08:00

io_test.py

Speed up convex-convex collisions.

2024-03-06 21:48:51 -08:00

io.py

Add cameras to MJX. Fixes #1422

2024-02-27 14:17:21 -08:00

math_test.py

Add MJX and bump version number to 3.0.0.

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

math.py

Unrolled matmul for small matrices in MJX improves steps/sec by 10-20%.

2024-02-07 10:48:45 -08:00

mesh_test.py

Speed up convex-convex collisions.

2024-03-06 21:48:51 -08:00

mesh.py

Speed up convex-convex collisions.

2024-03-06 21:48:51 -08:00

passive_test.py

Cleanup MJX tests and migrate them to put_model/put_data.

2023-12-05 23:08:05 -08:00

passive.py

Cleanup MJX tests and migrate them to put_model/put_data.

2023-12-05 23:08:05 -08:00

ray_test.py

Add naive ray mesh implementation.

2024-01-16 16:29:11 -08:00

ray.py

Add naive ray mesh implementation.

2024-01-16 16:29:11 -08:00

scan_test.py

Cleanup MJX tests and migrate them to put_model/put_data.

2023-12-05 23:08:05 -08:00

scan.py

Add cameras to MJX. Fixes #1422

2024-02-27 14:17:21 -08:00

smooth_test.py

Add cameras to MJX. Fixes #1422

2024-02-27 14:17:21 -08:00

smooth.py

Merge pull request #1447 from jaro-sevcik:mjx-factor-m-parallel

2024-02-28 22:31:01 -08:00

solver_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

solver.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

support_test.py

Add crude broadphase impl.

2024-02-29 17:12:58 -08:00

support.py

Add crude broadphase impl.

2024-02-29 17:12:58 -08:00

test_util.py

Remove TEST_FILES from MJX test_util.py.

2024-01-19 11:15:06 -08:00

types.py

Speed up convex-convex collisions.

2024-03-06 21:48:51 -08:00