This repo contains the code for my blog post on Visualizing Tensor Operations with Factor Graphs.
The code uses manimnx, a package to help interface manim with networkx to create the animations.
factor_graph.pycontains code which purely deals with computations and representing factor graphs using networkx.fg_anim.pyhas a few helper functions specific to animating factor graphs.gifs.pyhas code for each of the gifs/animations in the post.