torch>=2.0.0
numpy>=1.24.0

[dev]
pytest>=7.0.0
black>=23.0.0
isort>=5.12.0

[jax]
jax>=0.4.0
jaxlib

[ode]
torchdiffeq>=0.2.3

[thrml]
thrml

[vae]
diffusers>=0.21.0
transformers>=4.30.0
