--pre

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

# === Core Dependencies ===
jax==0.4.29
jaxlib==0.4.29+cuda12.cudnn91
flax
optax
chex
numpy
scipy

tensorflow
tensorflow-probability
distrax

mujoco==3.2.0
brax==0.12.0
gymnax
gym
craftax
