Does Pytorch-Lightning support the task of federated learning?

[marsggbo]

Does Pytorch-Lightning support the task of federated learning? If so, any example provided? Thanks!

[kaushikb11]

@tchaton Your take?

[tchaton]

Hey @marsggbo,

We worked on a first integration with PySyft. Here is the notebook: https://github.com/OpenMined/PySyft/blob/dev/packages/syft/examples/duet/mnist_lightning/MNIST_Lightning_Syft_Data_Scientist.ipynb

Only works for CPU right now.

They are currently working on Plan which make remote training way faster.

We will resume our collaboration in 1-2 month time.

Best,
T.C

[Geeks-Sid]

Any updates on this @tchaton ?

[tchaton]

Hey @Geeks-Sid,

There is no current plan to push this further. If you think there is a strong desire from the community, please open a feature request on PySyft.

Best,
T.C