Pytorch traning on android/iOS devices

Is there pytorch.js library (similar to tensorflow.js) to train models directlly on mobile edevices?

Hello Rohit! Unfortunately there’s isn’t a JS library for on-device training available at this point in time. However, you could train models directly using the C++ API since the full-jit interpreter can run on mobile platforms.

Thanks Dhruv.

Keep this thread updated when pyTorch decides to have one to compate with TF.