__init__.py 142 B

1234
  1. import torch
  2. if torch.distributed.rpc.is_available():
  3. from .api.remote_module import RemoteModule
  4. from .functional import * # noqa: F403