ml-agents/mlagents/torch_utils/__init__.py

from mlagents.torch_utils.torch import torch as torch  # noqa
from mlagents.torch_utils.torch import nn  # noqa
from mlagents.torch_utils.torch import set_torch_config  # noqa
from mlagents.torch_utils.torch import default_device  # noqa