absl-py
numpy>=1.18
opt_einsum
scipy>=1.2.1

[cpu]
jaxlib==0.1.71

[cuda102]
jaxlib==0.1.71+cuda102

[cuda111]
jaxlib==0.1.71+cuda111

[minimum-jaxlib]
jaxlib==0.1.69

[tpu]
jaxlib==0.1.71
libtpu-nightly==0.1.dev20210809
requests
