Code

Add cuda profiler hook to paxml#1089

Code

Open

shawnwang18 wants to merge 1 commit intoNVIDIA:mainNVIDIA/JAX-Toolbox:mainfrom shawnwang18:shawnw/add_paxml_patchshawnwang18/JAX-Toolbox:shawnw/add_paxml_patchCopy head branch name to clipboard

+1

Commits

Commits on Oct 9, 2024

• Add cuda profiler hook to paxml

  shawnwang18

  committed
  8f667e9

  Copy full SHA for 8f667e9