Guys, importing pytorch like that seems to work for me:
    implementation ('org.pytorch:pytorch_android:1.5.0'){
     exclude module: 'fbjni-java-only'
}
    implementation ('org.pytorch:pytorch_android_torchvision:1.5.0'){
     exclude module: 'fbjni-java-only'
}
    implementation ('org.pytorch:pytorch_android_fbjni:1.5.0'){
     exclude module: 'fbjni-java-only'
}
    implementation('com.facebook.react:react-native:+'){
     exclude module: 'fbjni-java-only'
}


I also have
android {
        packagingOptions {
	pickFirst '**/*.so'
}
}

so that it won’t go crazy over those multiple libfbjni.so I’m getting for some reason. Hope that helps!