Most linear algebra from numpy should be include with PyTorch already, but if you’re looking for a GPU supported version of sklearn, try skorch! GitHub - skorch-dev/skorch: A scikit-learn compatible neural network library that wraps PyTorch</t 2