ray-skorch - distributed PyTorch on Ray with sklearn API

tl;dr: train PyTorch models on large tabular datasets with a scikit-learn (skorch) API

Hi r/datascience,

I'm the principal author of ray-skorch, a library that lets you run distributed PyTorch training on large-scale datasets while providing a familiar, scikit-learn compatible skorch API, integrating well with the rest of the scikit-learn ecosystem.

Under the hood, ray-skorch uses Ray Train for distributed PyTorch training and Ray Data for handling and shuffling large datasets.

ray-skorch works only with tabular data. Currently, …

api datascience distributed pytorch sklearn