def METHOD_NAME(
    framework_version, ecr_image, sagemaker_regions, instance_type, dist_gpu_backend
):
    instance_type = instance_type or "ml.p3.8xlarge"
    training_group = InstanceGroup("train_group", instance_type, 2)
    function_args = {
        "framework_version": framework_version,
        "instance_groups": [training_group],
        "dist_backend": dist_gpu_backend,
        "use_inductor": True,
    }
    invoke_pytorch_helper_function(
        ecr_image, sagemaker_regions, _test_mnist_distributed, function_args
    )