implementation 'org.pytorch:pytorch_android_lite:1.9.0'
    implementation 'org.pytorch:pytorch_android_torchvision:1.9.0'

works for me on the ImageSegmentation app 6. It should just fetch libpytorch_jni_lite.so and I’m not sure why the error message shows that it still looks for libpytorch_jni.so. It looks like some cache issue to me.