numpy>=1.19.5

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

[jax]
jax>=0.2.20
jaxlib>=0.1.71
