https://github.com/blue-season/pywarm
PyWarm is a lightweight, high-level neural network construction API for PyTorch. It enables defining all parts of NNs in the functional way.
I don't see much of a reason you want to clutter your code with that, if you need a wrapper around pytorch use fastai not some obscure library nobody has ever heard about.
Thank god you weren’t around when keras was released.
Use fastai if you want your code to be unreadable, unmaintable and prone to mysterious bugs.