pytorch_lightning>=1.0.0
matplotlib
torchvision
pandas
xgboost>=1.1.0
einops
numpy
torch>=1.5.1
rebound>=3.14.0
scikit-learn
scipy
