numpy>=1.20
scipy>=1.7
torch
ninja

[all]
torch
jax
jaxlib

[cuda]
torch

[cuda12]
nvidia-cufft-cu12

[cuda13]
nvidia-cufft<13,>=12

[datasets]
requests>=2.25
tqdm>=4.60

[jax]
jax
jaxlib

[torch]
torch
