zombie-einstein

icon indicating a website link https://zombie-einstein.github.io/

icon location Research engineer interested in machine-learning and agent-based modelling Take a look at my blog https://zombie-einstein.github.io/

Results 1 issues of zombie-einstein

Add JaxPR-Viz

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

revisit