GitHub repository: ASKabalan/jax-hpc-profiler
Primary language: Python
Description: A convinient python script to plot performance graphs for HPC JAX codes