jax>=0.10.2
safetensors>=0.8.0
