jaxtyping<0.3,>=0.2
torch~=2.0
torchaudio~=2.0
numpy<2.0.0
scipy<1.15,>=1.9.0
