awesome-jax icon indicating copy to clipboard operation
awesome-jax copied to clipboard

Published 20 hours ago • verified publisher icon n2cholas

Add JaxPR-Viz

Open zombie-einstein opened this issue 9 months ago • 0 comments

I created this library designed to visualise JAX computation graphs from JaxPR, and to be sightly higher level than the built in HLO visualiser.

zombie-einstein avatar Sep 19 '23 11:09 zombie-einstein