tito

TinyTorch has you rebuild PyTorch's API in pure Python inside 4GB of RAM

Twenty modules in four tiers, a CLI called tito, and a prerequisite list of Python plus NumPy. The post on pytorch.org argues the payoff is new hires who can read production autograd on arrival, at a cost of two to three weeks.