Running TAPNET in Windows?

I am trying to run TAPNET in Windows 11 with Anaconda3-2023.07-2-Windows-x86_64.exe but I still get a JAX related error message.
I already installed `jaxlib-0.4.11-cp311-cp311-win_amd64.whl` but it´s apparently not enough.

```
(base) C:\Users\ATC\tapnet>python ./experiment.py --config ./configs/tapir_config.py
Traceback (most recent call last):
File "C:\Users\ATC\tapnet\experiment.py", line 29, in
from jaxline import experiment
File "C:\ProgramData\anaconda3\Lib\site-packages\jaxline\experiment.py", line 30, in
from jaxline import utils
File "C:\ProgramData\anaconda3\Lib\site-packages\jaxline\utils.py", line 335, in
rng: jnp.DeviceArray,
^^^^^^^^^^^^^^^
File "C:\ProgramData\anaconda3\Lib\site-packages\jax\_src\deprecations.py", line 53, in getattr
raise AttributeError(f"module {module!r} has no attribute {name!r}")
AttributeError: module 'jax.numpy' has no attribute 'DeviceArray'

```

conda list includes this:
```
jax 0.4.14 pypi_0 pypi
jaxlib 0.4.11 pypi_0 pypi
jaxline 0.0.5 pypi_0 pypi

```

Can somebody tell me what I am doing wrong?