Repository: pytorch
Issue: NotImplementedError: The operator aten::native_group_norm_backward
Labels: triaged,enhancement,module: mps

### 🐛 Describe the bug

While running [this](https://huggingface.co/blog/annotated-diffusion) notebook on PyTorch-1.13.0 for Apple M1 chip, i've faced the issue above on follow line of code: `loss.backward()`.

Here's the full error log:
```
---------------------------------------------------------------------------
NotImplementedError                       Traceback (most recent call last)
Cell In [27], line 20
     17 if step % 100 == 0:
     18   print("Loss:", loss.item())
---> 20 loss.backward()
     21 optimizer.step()
     23 # save generated images

File /opt/homebrew/Caskroom/miniforge/base/envs/torch/lib/python3.9/site-packages/torch/_tensor.py:484, in Tensor.backward(self, gradient, retain_graph, create_graph, inputs)
    474 if has_torch_function_unary(self):
    475     return handle_torch_function(
    476         Tensor.backward,
    477         (self,),
   (...)
    482         inputs=inputs,
    483     )
--> 484 torch.autograd.backward(
    485     self, gradient, retain_graph, create_graph, inputs=inputs
    486 )

File /opt/homebrew/Caskroom/miniforge/base/envs/torch/lib/python3.9/site-packages/torch/autograd/__init__.py:191, in backward(tensors, grad_tensors, retain_graph, create_graph, grad_variables, inputs)
    186     retain_graph = create_graph
    188 # The reason we repeat same the comment below is that
    189 # some Python versions print out the first line of a multi-line function
    190 # calls in the traceback and some print out the last line
--> 191 Variable._execution_engine.run_backward(  # Calls into the C++ engine to run the backward pass
    192     tensors, grad_tensors_, retain_graph, create_graph, inputs,
    193     allow_unreachable=True, accumulate_grad=True)

NotImplementedError: The operator 'aten::native_group_norm_backward' is not current implemented for the MPS device. If you want this op to be added in priority during the prototype phase of this feature, please comment on https://github.com/pytorch/pytorch/issues/77764. As a temporary fix, you can set the environment variable `PYTORCH_ENABLE_MPS_FALLBACK=1` to use the CPU as a fallback for this op. WARNING: this will be slower than running natively on MPS.
```

### Versions

Collecting environment information...
PyTorch version: 1.13.0.dev20220908
Is debug build: False
CUDA used to build PyTorch: None
ROCM used to build PyTorch: N/A

OS: macOS 12.5.1 (arm64)
GCC version: Could not collect
Clang version: 14.0.0 (clang-1400.0.29.102)
CMake version: version 3.24.1
Libc version: N/A

Python version: 3.9.13 | packaged by conda-forge | (main, May 27 2022, 17:01:00)  [Clang 13.0.1 ] (64-bit runtime)
Python platform: macOS-12.5.1-arm64-arm-64bit
Is CUDA available: False
CUDA runtime version: No CUDA
GPU models and configuration: No CUDA
Nvidia driver version: No CUDA
cuDNN version: No CUDA
HIP runtime version: N/A
MIOpen runtime version: N/A
Is XNNPACK available: True

Versions of relevant libraries:
[pip3] numpy==1.23.2
[pip3] torch==1.13.0.dev20220908
[pip3] torchaudio==0.13.0.dev20220908
[pip3] torchvision==0.14.0.dev20220908
[conda] numpy                     1.23.2                   pypi_0    pypi
[conda] torch                     1.13.0.dev20220908          pypi_0    pypi
[conda] torchaudio                0.13.0.dev20220908          pypi_0    pypi
[conda] torchvision               0.14.0.dev20220908          pypi_0    pypi

cc @kulinseth @albanD @malfet @DenisVieriu97 @jhavukainen @razarmehr @abhudev