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

[cpu]
jaxlib==0.1.73

[cuda]
jaxlib==0.1.73+cuda11.cudnn82

[cuda11_cudnn805]
jaxlib==0.1.73+cuda11.cudnn805

[cuda11_cudnn82]
jaxlib==0.1.73+cuda11.cudnn82

[minimum-jaxlib]
jaxlib==0.1.69

[tpu]
jaxlib==0.1.73
libtpu-nightly==0.1.dev20211018
requests
