Interactive Visualization of PyTorch (JIT) models

March 16, 2022

PyTorch turned five years old recently. On this very day, 5 years ago, I joined the PyTorch forum and wrote my first post.

It has been very exciting to participate in the early development of PyTorch with a whole bunch of features and bug fixes. I have met many awesome people both online and in person. I have been on the receiving end of a truely eye-watering amount of kindness for which I cannot express enough gratitude. (You know who you are!)

To celebrate, I am sharing a an interactive model visualization in Jupyter Notebooks and GraphViz with you, based on an earlier post from two years ago that people appear to have found useful.

Here is a quick video of it in action:

And here is a link to the notebook.

Here is hoping that I will have the good fortune to work with many of you in the next five years!

PyTorch Training and Consulting

If you need really great PyTorch Training and/or Consulting, do get in touch! Email me at tv@lernapparat.de.

If you are near M√ľnster, like PyTorch and want to grab a coffee, I would be glad to hear what you are using it for.