src/transformers/models/esm/openfold_utils/rigid_utils.py/Rigid/cuda
class Rigid:    def cuda(self) -> Rigid:
        """
        Moves the transformation object to GPU memory

        Returns:
            A version of the transformation on GPU
        """
        return Rigid(self._rots.cuda(), self._trans.cuda())