# === Core Dependencies ===
jax[cuda12]
flax
optax
chex


distrax
kinetix-env==2.0.3
git+https://github.com/DramaCow/jaxued.git@0f8f1284677375b889e4f13a32c9617cd009f8c4
