jax>=0.4.20
flax>=0.7.0

[all]
zenbit[pallas-flax]>=0.2.0
torch>=2.0

[jax]

[pallas]
zenbit[pallas-flax]>=0.2.0

[pytorch]
torch>=2.0
