typing_extensions>=4.13
jsonargparse>=4.30

[datasets]
datasets>=2.18

[dev]
pytest>=8
pyright>=1.1.386
build>=1.2

[jaxtyping]
jaxtyping>=0.2

[torch]
torch>=2.0
