Yeah, certain operations and features aren’t implemented yet. Support is still in its early stages and more optimizations await.
You can check this GitHub issue to see which operators are not implemented, are being implemented, and which ones are implemented: https://github.com/pytorch/pytorch/issues/77764