If you mean the organization of the operator invocations into Modules, I think a good reference is some of the implementations in torchvision e.g., the resnet implementation here:
https://pytorch.org/vision/stable/_modules/torchvision/models/resnet.html#resnet18 1