Issues in running colab tutorial in computer with GPU

### The feature, motivation and pitch

I am getting following error

jax.errors.SimplifiedTraceback: For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
File "/home/robot/Documents/GPU_accelerated_work/main.py", line 253, in
state = jit_reset(jax.random.PRNGKey(0))
File "/home/robot/Documents/GPU_accelerated_work/main.py", line 167, in reset
data = self.pipeline_init(qpos, qvel)
File "/home/robot/miniconda3/envs/mj/lib/python3.10/site-packages/brax/envs/base.py", line 126, in pipeline_init
return self._pipeline.init(self.sys, q, qd, act, ctrl, self._debug)
File "/home/robot/miniconda3/envs/mj/lib/python3.10/site-packages/brax/mjx/pipeline.py", line 81, in init
data = _reformat_contact(sys, data)
File "/home/robot/miniconda3/envs/mj/lib/python3.10/site-packages/brax/mjx/pipeline.py", line 35, in _reformat_contact
data = data.replace(
File "/home/robot/miniconda3/envs/mj/lib/python3.10/dataclasses.py", line 1453, in replace
return obj.__class__(**changes)
TypeError: Data.__init__() got an unexpected keyword argument 'contact'

please help

### Alternatives

_No response_

### Additional context

_No response_