An extension to trace, extract, and measure ops from running PyTorch models - View it on GitHub