custom nn# https://pytorch.org/tutorials/beginner/examples_nn/polynomial_module.html https://d2l.ai/chapter_builders-guide/custom-layer.html https://scribe.rip/14ab6ac94b77 TBD