numpy
matplotlib
scikit-learn
scikeras
tensorflow
torch
torchvision