mujoco_playground

[jiwon99yoon]

Hello, it's a really hot project and thanks for open-source this project

I tried some training on mujoco_playground without using your code (madrona_mjx)
it was successful and then i tried to train ppo using vision model but i got a issue

I wanna train vision-based PPO policy (in mujoco_playground env : PandaPickCube (--vision = True)),
But the brax, jax version dependencies are quite confused between mujoco_playground and Madrona MJX

so I wonder which version of brax, jax, you use in local computer to train that env.

thanks a lot

what i found

Brax 0.12.4 + JAX 0.5.3: error in cuSolver
JAX 0.6.0+: can't serve some API in Madrona-MJX(they said in github)
JAX 0.5.x: Dependency gap between Brax and CUDA (12.xx)

[azimi99]

My setup currently uses Jax 0.5.3, Brax 0.12.4, and cuda 12.5.1
Make sure JAX and JAXLib versions match.

[jiwon99yoon]

@azimi99
yes i did. but if i install jax 0.5.3 and then install brax 0.12.4 -> then brax makes dependency to make jax 0.7.1 (automatically update : guess dependency issue) : how should i solve this?
(my setting : cuda 12.5.1, driver : 575.06.03, in mujoco_playground uv env)

would you mind if you upload all your packages version or dependency?