torch/nn/__init__.pyi

from .modules import * 
from .parameter import Parameter as Parameter
from .parallel import DataParallel as DataParallel
from . import init as init
from . import utils as utils
from . import functional as functional