src2/__init__.py

from . import data 
from . import nn
from . import optim
from . import zoo