pandas
scikit-learn
matplotlib
torch
lightning

[dev]
pytest
