numpy>=1.19.5

[:python_version < "3.7"]
dataclasses>=0.7

[jax]
jax>=0.1.71

[jaxlib]
jaxlib>=0.1.49
