run code with new version of brax and jax ?

jiawei-ren / diffmimic

[ICLR 2023] DiffMimic: Efficient Motion Mimicking with Differentiable Physics https://arxiv.org/abs/2304.03274

https://diffmimic.github.io/

Other

266 stars 21 forks source link

run code with new version of brax and jax ? #6

Closed ucacaxm closed 2 days ago

ucacaxm commented 3 months ago

Hi,

Do you think the code may run with new version of brax and jax ?

I have: brax 0.10.5 brax-jumpy 0.1.0 flax 0.8.4 jax 0.4.29 jax-cuda12-pjrt 0.4.29 jax-cuda12-plugin 0.4.29 jaxlib 0.4.29 jaxopt 0.8.3 optax 0.2.2 orbax-checkpoint 0.5.16

I tested but I have issue with jumpy. I can search but I'd like to know if it's worth it or if too much has changed in Brax and Jax ?

Thanks

Galleta12 commented 3 months ago

Hello First of all thank you for this amazing work.

I am working on something similar and I was wondering the same.

Also I was wondering how can we create our humanoid model in the new version of Brax. Because the humanoid model I think it is in json format, but when I checked on the latest brax version they use xml for their models. Also, does this can also work with PPO?

Thanks