numpy
matplotlib
pillow
pillow_heif
opencv-python
torch
torchvision
jax<0.11,>=0.9.1
tqdm
scipy
sympy
tabulate
stablehlo_coreml==0.1.6

[cuda]
jax[cuda13]<0.11,>=0.9.1

[cuda12]
jax[cuda12]<0.11,>=0.9.1

[cuda13]
jax[cuda13]<0.11,>=0.9.1

[dev]
armpit

[metal]
coremltools==9.1.dev1
