d7f131e086
Add ellipsoid plane.
...
Co-authored-by: Alessio Quaglino <quaglino@google.com >
PiperOrigin-RevId: 621208833
Change-Id: I270c1ae0cecd7dd2d9b41da4e4d0b6378b105aaf
2024-04-02 09:49:46 -07:00
d258d5e152
Check all edge pairs in convex-convex.
...
PiperOrigin-RevId: 620004680
Change-Id: I5d75c0fd21ed3d3e420eed762cfc5a7e04896032
2024-03-28 11:33:46 -07:00
2019d42ca6
Small change to testspeed file.
...
PiperOrigin-RevId: 618323795
Change-Id: I3a6307b8ae6d9af0d17c0ae85eae673d60112ddf
2024-03-22 17:04:49 -07:00
624becca52
Fix some more bugs in sphere/capsule-convex collisions.
...
PiperOrigin-RevId: 617646119
Change-Id: Ie0c99ccbe1dd2a18663d45a111e77fed4112e638
2024-03-20 15:46:13 -07:00
9b86329797
Add 01-10 humanoids for benchmarking.
...
PiperOrigin-RevId: 617318016
Change-Id: Ic5dc0623266d4d6bef6da97d03370010e2817a2a
2024-03-19 15:48:48 -07:00
b4694ee331
Fix bug in MJX viewer.
...
PiperOrigin-RevId: 617297574
Change-Id: I34fb797db47308a8b2f29a6c54f2b59d77f84a32
2024-03-19 14:36:15 -07:00
e95159b4f6
Small fix to mjx tutorial.
...
PiperOrigin-RevId: 616235636
Change-Id: I38675c032bdca8f26ab0550896928d097ee86653
2024-03-15 14:00:28 -07:00
2386353b02
Throw an error for margin/gap with meshes.
...
PiperOrigin-RevId: 615937263
Change-Id: I3289cf1b1fec93385cc78a01e53866427453db59
2024-03-14 16:21:05 -07:00
02c62c118c
Fix bug where treedef for mjx.make_data doesn't match mjx.put_data.
...
PiperOrigin-RevId: 615928973
Change-Id: I08af97ac0300b9d51c1f4b46d48dde98d24197fa
2024-03-14 15:51:15 -07:00
e5811ce2b4
Further improve edge contact for capsule-convex.
...
PiperOrigin-RevId: 615900645
Change-Id: I7fd495b10b7b3121da7aa5ed15cde413d916bc84
2024-03-14 14:15:44 -07:00
f818a16ae4
Fix the bugfix for capsule-convex collisions.
...
PiperOrigin-RevId: 615809115
Change-Id: I183336630a76a9ada5b0730559b49779dd94f8e5
2024-03-14 09:41:33 -07:00
f2e107f7aa
Fix deep penetration for sphere/capsule-convex.
...
PiperOrigin-RevId: 615539264
Change-Id: I6a5aac0574ddcb39bae8aafaf63f9561ca69288b
2024-03-13 14:12:42 -07:00
a49f3f63fc
Remove disable_jit in test.
...
PiperOrigin-RevId: 614824680
Change-Id: I15f1441dfb1a70b98e5a221c0e15b06572860748
2024-03-11 16:03:40 -07:00
22cd96fbbc
Speed up convex-convex collisions.
...
PiperOrigin-RevId: 613450874
Change-Id: Ie66e0a078e2ff51a136b949dd9459847e1fe957d
2024-03-06 21:48:51 -08:00
106453fa2b
Updated version number to be (v3.1.4) ahead of current release (v3.1.3)
...
PiperOrigin-RevId: 613236845
Change-Id: I3ee759493f27cfa1e367c71e3e073b54738b73d4
2024-03-06 09:07:25 -08:00
a00f1b5f30
Tidy up Colab notebook files.
...
PiperOrigin-RevId: 613163325
Change-Id: Iec56375bb7631b8a3035f884b3aad6b54fb7b88b
2024-03-06 04:14:11 -08:00
224fbd00c8
Fix egregious bug in broadphase.
...
PiperOrigin-RevId: 612856292
Change-Id: I622ca00509c27f351ce5efefc139b2f2e141c8e9
2024-03-05 08:48:09 -08:00
419be4c605
Add crude broadphase impl.
...
PiperOrigin-RevId: 611643683
Change-Id: Id7816dadd7e9f1b8855268b815353c61540c4e0c
2024-02-29 17:12:58 -08:00
f9c46e2abb
Merge pull request #1447 from jaro-sevcik:mjx-factor-m-parallel
...
PiperOrigin-RevId: 611348280
Change-Id: I19ef25c732d43bba0b0a8c296b197948c69d5b52
2024-02-28 22:31:01 -08:00
803f0cb0d9
Clean up factor_m
2024-02-28 01:19:12 -08:00
a5ea87b0e6
Merge pull request #1446 from jaro-sevcik:mjx-solve-m-parallel
...
PiperOrigin-RevId: 610984251
Change-Id: Ifd69480643df236fe2d6b9f77d1b029d2650acff
2024-02-27 22:21:11 -08:00
2e43f18bee
Update barkour notebook.
...
PiperOrigin-RevId: 610906071
Change-Id: Ie67b86888e195bbd8a8e879547e65fbc2c864534
2024-02-27 16:09:20 -08:00
2067e2083a
Add cameras to MJX. Fixes #1422
...
PiperOrigin-RevId: 610870732
Change-Id: Iaa39bd00f56046c595dc961281ac23446ca5bdf2
2024-02-27 14:17:21 -08:00
7688e03eea
Remove unnecessary parens
2024-02-27 00:10:21 -08:00
29d6e10375
Update independent elements in solve_m in parallel
2024-02-25 20:27:28 +00:00
13f256dd19
Parallel factor_m
2024-02-25 20:24:54 +00:00
1b93226fb0
To save memory, avoid loading meshes that will never be used for collisions.
...
PiperOrigin-RevId: 607770456
Change-Id: I34c4a34d1820a9fd88d2ebeaffec9a64637d2614
2024-02-16 12:40:35 -08:00
6a346c42e8
Adds support for mjOption.impratio to MJX.
...
PiperOrigin-RevId: 607456034
Change-Id: I7f6549a4332fd2bc0331d76a8906b971ce28dbcf
2024-02-15 14:28:36 -08:00
d561995c03
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.
...
PiperOrigin-RevId: 607096855
Change-Id: I2e14b9b63e676b4ac3ec2a12aa07316e1d019a79
2024-02-14 13:57:55 -08:00
f50a44e193
Allow jax config to choose precision.
...
PiperOrigin-RevId: 606765036
Change-Id: I36977851615700507eb2e83cc6d2d1ff8255be62
2024-02-13 15:24:02 -08:00
47bb4a828e
Fixes to the MJX Euler integrator and Newton solver, and test improvements to catch future bugs like this.
...
PiperOrigin-RevId: 606389094
Change-Id: I50d0143f338fe29a823c930dba82995ce8a64ce2
2024-02-12 15:46:27 -08:00
805d862d5f
Fix manifold point function.
...
PiperOrigin-RevId: 606366838
Change-Id: I8cf031f6f0fd0173777e0a9bb6d11c0c77c617ec
2024-02-12 14:32:24 -08:00
e77c3cb258
Update MJX viewer to use mjx.get_data_into and fix bad function parameters.
...
PiperOrigin-RevId: 605461208
Change-Id: I7a4f6a1ed92382e522d50224e7b5cd73d8842750
2024-02-08 16:32:27 -08:00
4933a2c7b6
Improve performance of getting and putting MJX device data by reducing cost of ncon.
...
PiperOrigin-RevId: 605454760
Change-Id: I5c52a84c12ab67d042fe43f6ada284b51a762294
2024-02-08 16:06:58 -08:00
3e2d62c248
Improve performance of getting and putting MJX device data with large numpy arrays.
...
ndarray.tobytes() is faster than converting arrays to tuples by orders of magnitude.
PiperOrigin-RevId: 605437697
Change-Id: I7d42c10e178b1aec80ff38a82e863e0701c15e62
2024-02-08 15:02:21 -08:00
3baab3784a
Update MJX notebook to work with brax 0.10.0.
...
PiperOrigin-RevId: 605105172
Change-Id: Icd209ada7977439ed7d5213588412d37dd54ffc0
2024-02-07 14:49:17 -08:00
ea69f20b43
Unrolled matmul for small matrices in MJX improves steps/sec by 10-20%.
...
PiperOrigin-RevId: 605033188
Change-Id: I9da723d01e1f6511abaf4522b4c8f829a8091772
2024-02-07 10:48:45 -08:00
19c41aa965
Merge pull request #1401 from driesmarzougui:main
...
PiperOrigin-RevId: 604771114
Change-Id: I9389828e9b588c22eb24a4074a23283943ca748d
2024-02-06 14:49:35 -08:00
38cbc9e543
Improve MJX performance on NVIDIA GPU by enabling Triton GEMM.
...
PiperOrigin-RevId: 604691509
Change-Id: Iba11a9e1b79848624a1ef6eb4e61e2093d36e662
2024-02-06 10:31:49 -08:00
f763254a00
Updating Mujoco Release version number to v3.1.3
...
PiperOrigin-RevId: 604459311
Change-Id: I766f704c28740cf751cb935bb48fb6954fccf024
2024-02-05 15:59:20 -08:00
ed36956b0d
fix https://github.com/google-deepmind/mujoco/issues/1379
2024-02-05 10:59:08 +01:00
92c161a61b
Add ncon to __init__.
...
PiperOrigin-RevId: 603468404
Change-Id: I20c622b8f6f84ecacb9f3cdd26ac9560cf2788be
2024-02-01 13:38:23 -08:00
8be966cdf9
Avoid NaN in gradient of where. Fixes #1349 .
...
PiperOrigin-RevId: 602461299
Change-Id: I7c976d3b7262027bdc63967dcc00674fb2802e70
2024-01-29 12:09:29 -08:00
508669a9f2
Raise not implemented for frictionloss for now. #1344
...
PiperOrigin-RevId: 602453907
Change-Id: I0ea506b54f96c4fadc0f29f9ed0a21b0b6a7e5d9
2024-01-29 11:45:36 -08:00
a1e5ede266
Allow Brax data structures to inherit from MJX data structures.
...
PiperOrigin-RevId: 601290735
Change-Id: I6dad1b6d7e6b3e71acecc36189ab59cc81ab90b4
2024-01-24 18:03:17 -08:00
e59605135a
Make benchmarks and viewer more accessible.
...
- Drop google_benchmark module dependency, wasn't really using it.
- Move benchmark models into test_data
- Rename benchmark => testspeed, make default output more like MJ testspeed.
- Update viewer to use new io functions and add get_data_into.
- Add "mjx-testspeed" and "mjx-viewer" bin scripts
PiperOrigin-RevId: 601277135
Change-Id: I782f666c58a96ac1a0a1c5ffc8ecf29d9fcd810f
2024-01-24 17:04:02 -08:00
0a7be1732c
Adds support for explicit dense/sparse mass matrices to MJX.
...
This increases performance, particularly for the Newton solver on TPU.
PiperOrigin-RevId: 600696483
Change-Id: If69bb9a2e21ba8dad6ca23f093ce7b7ceae644ff
2024-01-23 00:32:07 -08:00
5cbaa23388
Remove TEST_FILES from MJX test_util.py.
...
PiperOrigin-RevId: 599886327
Change-Id: I4d166e09db22fffc959dbe34fa3469bd4a1b9f84
2024-01-19 11:15:06 -08:00
a02fc405af
Add naive ray mesh implementation.
...
PiperOrigin-RevId: 598993152
Change-Id: I708505bf20a89d5a4840451a11b99771195003e6
2024-01-16 16:29:11 -08:00
ee922be393
Benchmark improvements.
...
- Use multiple devices if present
- FLAGS more agnostic to diverse platforms
PiperOrigin-RevId: 598990299
Change-Id: Ifbd454ca521a788324f2b62f1c79e7d4b8305a30
2024-01-16 16:17:56 -08:00
Baruch Tabanpour