torch>=2.0.0
opt_einsum
pytest
functorch
hydra-core
matplotlib
pandas