absl-py
numpy>=1.20
opt_einsum
scipy>=1.5
typing_extensions
etils[epath]

[australis]
protobuf<4,>=3.13

[ci]
jaxlib==0.3.15

[cpu]
jaxlib==0.3.20

[cuda]
jaxlib==0.3.20+cuda11.cudnn82

[cuda11_cudnn805]
jaxlib==0.3.20+cuda11.cudnn805

[cuda11_cudnn82]
jaxlib==0.3.20+cuda11.cudnn82

[minimum-jaxlib]
jaxlib==0.3.15

[tpu]
jaxlib==0.3.20
libtpu-nightly==0.1.dev20220928
requests
