jax>=0.9.0
jaxlib>=0.9.0

[dev]
pytest

[export]
coremltools>=8.0
