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!