You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
[JAX] Bug Fix: Softmax FFIs with correct Encapsulates#1375
Merged
phu0ngng merged 4 commits intoNVIDIA:mainNVIDIA/TransformerEngine:mainfrom phu0ngng:bugfix_ffi_softmaxphu0ngng/TransformerEngine:bugfix_ffi_softmaxCopy head branch name to clipboardDec 14, 2024
![pre-commit-ci[bot]](https://avatars.githubusercontent.com/in/68672?v=4&size=32){kind=avatar}