pytorch

38e96405 - Reland python ops (#99170)

Reland python ops (#99170) Waiting for the revert to land. Pull Request resolved: https://github.com/pytorch/pytorch/pull/99170 Approved by: https://github.com/albanD