# Core dependencies for both JAX and NumPy implementations numpy>=1.20.0 gymnasium>=0.27.0 # JAX implementation dependencies jax>=0.4.3 jaxlib>=0.4.3 chex>=0.1.6 flax>=0.6.4 gymnax>=0.0.6 # Optional dependencies for visualization matplotlib>=3.5.0