src/transformers/models/esm/openfold_utils/rigid_utils.py/Rigid/stop_rot_gradient
class Rigid:    def stop_rot_gradient(self) -> Rigid:
        """
        Detaches the underlying rotation object

        Returns:
            A transformation object with detached rotations
        """
        return self.apply_rot_fn(lambda r: r.detach())