Use joblib from sklearn