absl-py>=0.7.1
jmp>=0.0.2
numpy>=1.18.0
tabulate>=0.8.9

[flax]
flax>=0.7.1

[jax]
jax>=0.4.28
jaxlib>=0.4.28
