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