As @cccclai mentioned we are building light-weighted runtime on both Android and iOS. It’s not super difficult to add training capability. Actually we have tests running in test/cpp/jit/test_lite_trainer.cpp, for functionality only. We’ll let you know when we put the necessary components into Android build.