If the app use ‘org.pytorch:pytorch_android:1.9.0’, Module.load is the correct api to load.
If the app use ‘org.pytorch:pytorch_android_lite:1.9.0’, LiteModuleLoader.load is the correct api to load.
Is it possible for you to use LiteModuleLoader.load in Java Api + pytorch_android_lite in build.gradle?