torch>=1.11.0
numpy>=1.21.6
scikit-learn>=0.24.2
pandas>=1.3.5
scipy>=1.5.4
functorch~=0.1.1
