https://github.com/askabalan/jax-hpc-profiler • Rank 6.7 • Science 13%

A convinient python script to plot performance graphs for HPC JAX codes