torch<2.0.0,>=1.4.0
numpy<2.0.0,>=1.18.2
scipy<2.0.0,>=1.4.1
numba<0.49.0,>=0.48.0
scikit-learn<0.24.0,>=0.23.0
tensorboard<3.0.0,>=2.2.0
matplotlib<4.0.0,>=3.0.0
