the state of the art pytorch model should have the following capabilities:

all of its layers and operations should be compatible with the native mobile devices (Android and iOS)
for Android: pytorch -> ONNX -> TensorFlow -> tflite

for iOS: pytorch -> coremltools -> coreml

model should able to use hardware acceleration on mobile devices (if the mobile supports)

Any help will be very much appreciated 
Also, I’ve 2 months’ time frame for this. So, I’ll keep posting my progress here itself (just in case, if it can be of help to anyone).