Is there a library or function to create a manuscript ready schematic of a pytorch model?

As the title suggests, I am wondering if anyone is aware of a library that will create a schematic of your pytorch model?

This is a snip from google of Alexnet, or something similar:

image