absl-py
numpy>=1.12
opt_einsum

[cpu]
jaxlib>=0.1.65

[cuda101]
jaxlib==0.1.65+cuda101

[cuda102]
jaxlib==0.1.65+cuda102

[cuda110]
jaxlib==0.1.65+cuda110

[cuda111]
jaxlib==0.1.65+cuda111

[cuda112]
jaxlib==0.1.65+cuda112

[minimum-jaxlib]
jaxlib==0.1.65
