src/transformers/models/esm/openfold_utils/rigid_utils.py/Rigid/device
class Rigid:    def device(self) -> torch.device:
        """
        Returns the device on which the Rigid's tensors are located.

        Returns:
            The device on which the Rigid's tensors are located
        """
        return self._trans.device