torch>=2.0
zadapter>=0.1.0

[pallas]
jax>=0.7.0

[pallas-flax]
jax>=0.7.0
flax>=0.8.0

[tpu]
torch_xla>=2.0
