numpy>=1.20
opt_einsum
scipy>=1.5

[australis]
protobuf<4,>=3.13

[ci]
jaxlib==0.4.1

[cpu]
jaxlib==0.4.2

[cuda]
jaxlib==0.4.2+cuda11.cudnn86

[cuda11_cudnn82]
jaxlib==0.4.2+cuda11.cudnn82

[cuda11_cudnn86]
jaxlib==0.4.2+cuda11.cudnn86

[minimum-jaxlib]
jaxlib==0.4.1

[tpu]
jaxlib==0.4.2
libtpu-nightly==0.1.dev20230124
requests
