numpy>=1.18
absl-py
opt_einsum

[cpu]
jaxlib==0.1.70

[cuda101]
jaxlib==0.1.70+cuda101

[cuda102]
jaxlib==0.1.70+cuda102

[cuda110]
jaxlib==0.1.70+cuda110

[cuda111]
jaxlib==0.1.70+cuda111

[minimum-jaxlib]
jaxlib==0.1.69

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