Releases: zombie-einstein/jaxpr-viz
Releases · zombie-einstein/jaxpr-viz
v0.1.7 Labelling Bug Fix
Fix a bug where graph nodes were incorrectly assigned ids and labels, introduced due to new naming convention of nodes in JaxPRs
Styling Tweaks
Various rendered graph styling tweaks
While Support
Add support to render graphs containing while
primitives
Switch Fixes
Various fixes to switch cases
Scan support
Improve rendering of 'scan' primitives
Update dependencies
Updates scipy version due to security issue
Add conditonal primitive support
Add support for generating sub-graphs representing conditional JAX functions, and a general restructure of the codebase.
v.0.1.0
v0.1.0 First Commit