carabiner-tools>=0.0.5.post3
numpy
tqdm

[dev]
pytest>=7.4
pytest-cov
duvida[jax,torch]

[jax]
jax>=0.5.3

[jax_cuda12]
jax[cuda12]>=0.5.3

[jax_cuda12_local]
jax[cuda12_local]>=0.5.3

[torch]
torch>=2.4
