GitHub repository: zombie-einstein/jaxpr-viz
Primary language: Python
Description: Jaxpr Visualisation Tool