numpy
dm_tree
tensorflow==2.8.*
tensorflow_io
tensorflow_probability==0.16.*
dm_sonnet
wandb
cpprb
absl-py
gymnasium
requests

[jax]
flashbax
optax
jax
flax
orbax-checkpoint
