models/lib/nn/__init__.py

from .modules import *
from .parallel import (UserScatteredDataParallel, async_copy_to,
                       user_scattered_collate)