brax>=0.9.0
chex
colorama
craftax
distrax @ git+https://github.com/google-deepmind/distrax  # distrax release doesn't support jax > 0.4.13
flashbax @ git+https://github.com/instadeepai/flashbax
flax
gymnax @ git+https://github.com/RobertTLange/gymnax
huggingface_hub
hydra-core==1.3.2
id-marl-eval @ git+https://github.com/instadeepai/marl-eval
jax>=0.4.10
jaxlib
jaxmarl
joblib==1.4.0
jumanji @ git+https://github.com/instadeepai/jumanji.git
mctx
neptune
numpy
omegaconf
optax
optuna==3.6.1
pgx
protobuf==3.20.1
rlax
simplejson==3.18 # prevents a nan error in neptune uploading
tdqm
tensorboard_logger
tensorflow_probability
xminigrid @ git+https://github.com/corl-team/xland-minigrid.git@main
