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

[:python_version < "3.8"]
typing_extensions

[jax]
jax>=0.1.71

[jaxlib]
jaxlib>=0.1.49
