numpy
scikit-learn
torch

hypothesis