absl-py
numpy>=1.12
opt_einsum

[cpu]
jaxlib>=0.1.64

[cuda-101]
jaxlib==0.1.64+cuda101

[cuda-102]
jaxlib==0.1.64+cuda102

[cuda-110]
jaxlib==0.1.64+cuda110

[cuda-111]
jaxlib==0.1.64+cuda111

[cuda-112]
jaxlib==0.1.64+cuda112

[minimum-jaxlib]
jaxlib==0.1.64
