One way is via comments, like this:

PyWarm, the functional API to build neural networks for PyTorch https://github.com/blue-season/pywarm