jax>=0.2.20
jaxlib>=0.1.69
numpy

[:python_version < "3.8"]
importlib-metadata
