Does Pytorch support training on android devices ( with java codes)

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: