Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
TransformerEngine
Repository
3ff0b8d4934f1459d2ca84a5cb312b9d678b1a5a
Switch branch/tag
TransformerEngine
docs
examples
quickstart_jax.ipynb
Find file
Blame
History
Permalink
Change Flax MHA to DPA to remove the duplicated QKV projection step (#2429)
· 3ff0b8d4
Teddy Do
authored
Nov 26, 2025
Signed-off-by:
tdophung
<
tdophung@nvidia.com
>
3ff0b8d4
quickstart_jax.ipynb
31 KB
Edit
Web IDE
Replace quickstart_jax.ipynb
×
Attach a file by drag & drop or
click to upload
Commit message
Replace quickstart_jax.ipynb
Replace file
Cancel
A new branch will be created in your fork and a new merge request will be started.
tdophung <tdophung@nvidia.com>