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.
Keep this thread updated when pyTorch decides to have one to compate with TF.