--pre

--find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

# === Core Dependencies ===
jax[cuda12]==0.4.33
jaxmarl~=0.1.0
flax==0.8.5
safetensors
chex
brax==0.10.3
mujoco==3.1.3
matplotlib
pillow
scipy<=1.12
gymnax
optax
distrax
flashbax==0.1.0
