It's also officially in beta as of 1.13 https://pytorch.org/blog/PyTorch-1.13-release/#beta-support-...