Is there a way to get the .xml file of the locomotion soccer environment? Run environment in GPU?

[StefanoBraghetto]

I want to create a project from this environment: soccer
But I can't get with the .xml. My final objective is to run it in the GPU so I can run multiple instance of it at the same time. I was reading that this is possible with mujoco_mjx and I could load the environment doing:

mj_model = mujoco.MjModel.from_xml_path(
        (path / 'name_file.xml').as_posix())

please help!

[sibocw]

I think #429 gives the answer, unless the developers have since decided to retrofit MJX?

[kevinzakka]

@sibocw @StefanoBraghetto I have an implementation of dm_control suite that is fully based on MJX and will be open-sourced very soon.

[sibocw]

@kevinzakka Awesome! Looking forward :)