__init__.py

from . import (
    bert,
    gpt2,
    hanging_param_model,
    inline_op_model,
    nested_model,
    repeated_computed_layer,
    resnet,
    simple_net,
)
from .utils import run_fwd_bwd