I recommend using VS Code with LaTeX Workshop for easily visualizing changes to figures. In preview mode, once you run the code to regenerate the tex file for the figure, the figure should update automatically in the pdf preview window.
Check the examples:
- tree: [py] [tex] [pdf]
- transformer: [py] [tex] [pdf]
- xlnet: [py] [tex] [pdf]
- lda: [py] [tex] [pdf]
- deep_architect: [py] [tex] [pdf]
- Easy GIF generation
- Multiple output formats (png, svg, ...)
- 3D plotting
- More 2D plotting primitives for common plots (bar, line, scatter, ...)
- Slide deck creation (i.e., an alternative to Beamer)