absl-py
dm-haiku
jax>=0.4.4
jaxlib
msgpack
optax
requests
scipy
