jax>=0.10.0
jaxlib>=0.10.0

[dev]
pytest

[export]
coremltools>=8.0
stablehlo-coreml>=0.1.3
