src/__init__.py

from .models import *
from .metrics import *
from .datasets import *
from .losses import *