PyTorch Lightning Integration

dmmiller612 / sparktorch

Train and run Pytorch models on Apache Spark.

MIT License

339 stars 44 forks source link

Open tchaton opened 2 years ago

tchaton commented 2 years ago

Very good work there. I wondered if you would be interested in adding support for PyTorch Lightning.

This would enable users to faster train their models and remove some boilerplate. Plus, I believe this could be nicely packaged under a SparkTrainer.

Best, T.C