src/models/__init__.py

from .gpt import GPT
from .block import Block
from .attention import CausalSelfAttention
from .mlp import MLP