diff --git a/.github/container/manifest.yaml b/.github/container/manifest.yaml index e9d30a3bc..17e454d89 100644 --- a/.github/container/manifest.yaml +++ b/.github/container/manifest.yaml @@ -40,6 +40,7 @@ paxml: patches: pull/46/head: file://patches/paxml/PR-46.patch # adds Transformer Engine support pull/93/head: file://patches/paxml/PR-93.patch # exposes TE DPA argument in Grok to make it configurable from the command line + pull/98/head: file://patches/paxml/PR-98.patch # Add cuda profiler hook function call to paxml praxis: url: https://github.com/google/praxis.git mirror_url: https://github.com/nvjax-svc-0/praxis.git