Wonder if pytorch support training on android devices to perform federated learning in practice
At the moment we do not support training in pytorch mobile. We are looking on it. Stay tuned.
Regarding Java code, as long as we can build the C++ API for Android, we can use that from Java with the JavaCPP Presets for PyTorch: