Commit Graph

26 Commits

Author SHA1 Message Date
Baruch Tabanpour 43a7493d17 Fix embarrassing bug with ckpt.
PiperOrigin-RevId: 686679906
Change-Id: I61126ce94214b5f7b25b31e343a243d1a1dbb9f4
2024-10-16 16:23:54 -07:00
Copybara-Service 39af655c1e Merge pull request #1939 from goncalog:update-humanoid-notebook
PiperOrigin-RevId: 683668357
Change-Id: Id4a65197efc7d6d36e3e1d9885691b432d6114eb
2024-10-08 10:10:20 -07:00
Goncalo Gordo 76a8d01eb9 revert physics_steps_per_control_step 2024-10-05 13:25:47 +01:00
Goncalo Gordo b08d6f4724 revert the solver params 2024-10-05 13:23:47 +01:00
Baruch Tabanpour 0dde0ba4ba Fix external import in colab.
PiperOrigin-RevId: 682514874
Change-Id: I6157c2613464d2f5d3c23a73248913f1a240a9f1
2024-10-04 18:12:43 -07:00
Baruch Tabanpour 0887f54307 Add Franka Panda env to MJX colab tutorial.
PiperOrigin-RevId: 682506877
Change-Id: Ifde31e993f63baa783e973c496e5bd7bd6465393
2024-10-04 17:36:49 -07:00
Goncalo Gordo 5faff169b7 Update simulation params in MJX's tutorial 2024-08-29 14:39:08 +01:00
Goncalo Gordo ee731506d7 Update params in humanoid's training 2024-08-29 14:37:38 +01:00
Baruch Tabanpour c136f3d6e8 Fix barkour path.
PiperOrigin-RevId: 641093437
Change-Id: I92b8193c8458461e1a1f80b1e4c099067ed2099a
2024-06-06 19:09:58 -07:00
Baruch Tabanpour fb8bf206d9 Update MJX colab with hfield example.
PiperOrigin-RevId: 641061897
Change-Id: Ia1002173adce7b7b12203857abed18568ae236f1
2024-06-06 16:43:34 -07:00
Baruch Tabanpour 2a9067a6b3 Replace dt with opt.timestep.
PiperOrigin-RevId: 636984197
Change-Id: I2b5eaae0c5e5be840f93c4fe8f725884959703cd
2024-05-24 11:35:22 -07:00
Google DeepMind 8b7f1094f1 Fix jax deprecation warning for jax.tree_map.
PiperOrigin-RevId: 626092308
Change-Id: I78145efe62ae118726c1755856d7cb9b1abd988a
2024-04-18 11:13:35 -07:00
Baruch Tabanpour e95159b4f6 Small fix to mjx tutorial.
PiperOrigin-RevId: 616235636
Change-Id: I38675c032bdca8f26ab0550896928d097ee86653
2024-03-15 14:00:28 -07:00
Nimrod Gileadi a00f1b5f30 Tidy up Colab notebook files.
PiperOrigin-RevId: 613163325
Change-Id: Iec56375bb7631b8a3035f884b3aad6b54fb7b88b
2024-03-06 04:14:11 -08:00
Baruch Tabanpour 2e43f18bee Update barkour notebook.
PiperOrigin-RevId: 610906071
Change-Id: Ie67b86888e195bbd8a8e879547e65fbc2c864534
2024-02-27 16:09:20 -08:00
Baruch Tabanpour 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
Erik Frey 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
Baruch Tabanpour 721e2d5589 Update tutorial notebook.
PiperOrigin-RevId: 595746621
Change-Id: Ie0a4ea01d4f50491b14713f12ea8a4fe6877f890
2024-01-04 10:45:00 -08:00
Tyler Lindberg b82c382c53 Use positional arguments with jp.where 2023-12-12 16:28:36 -08:00
Baruch Tabanpour e7d637f375 Fix issue with foot vel in MJX notebook
PiperOrigin-RevId: 589276253
Change-Id: I2c8c24e42d037d8c59667364a298f8b669d779b6
2023-12-08 16:44:36 -08:00
Erik Frey 3c0a56c1e5 Adds support for Newton solver to MJX.
PiperOrigin-RevId: 578738033
Change-Id: Id0e68c7b961380cb1ed88b40d99f202c3fa8e228
2023-11-01 22:05:10 -07:00
Baruch Tabanpour a4588cee12 Fix https://github.com/google-deepmind/mujoco/issues/1123
PiperOrigin-RevId: 576159322
Change-Id: I29fc3df294af008f55324bc49c0f597d778e3392
2023-10-24 09:08:34 -07:00
Baruch Tabanpour e57b34303a Reformat notebook and add back sim2sim.
PiperOrigin-RevId: 575246433
Change-Id: I8627916e7dedd5cf64dab6e0151f41c3153e7a68
2023-10-20 09:53:53 -07:00
Baruch Tabanpour 9decb7389f Make humanoid default in colab.
PiperOrigin-RevId: 574925299
Change-Id: I619438b4c61b882db27c3984fb8c061ec0c39e4d
2023-10-19 10:38:09 -07:00
Baruch Tabanpour b989a85364 Remove block in colab.
PiperOrigin-RevId: 574595549
Change-Id: I3814a8ecb909a581d96b614edd99002c05de014b
2023-10-18 14:19:05 -07:00
Erik Frey 8f9c690c85 Add MJX and bump version number to 3.0.0.
Co-authored-by: Baruch Tabanpour <btaba@google.com>
PiperOrigin-RevId: 574327508
Change-Id: Ia9b62fbc929c6869dfcec87636b2e10d405a1060
2023-10-18 10:33:29 +01:00