jaxtyping>=0.3.2
numpy>=2.3.3
torch>=2.8.0
