Add CI testing execution of nsys-jax notebook.

NVIDIA / JAX-Toolbox

JAX-Toolbox

Apache License 2.0

234 stars 44 forks source link

Closed olupton closed 3 months ago

olupton commented 3 months ago

This reverts commit fa1956c3aa24624c7ef2a89bf5c7de647eaee08e and fixes some bugs.