Skip to content

Navigation Menu

Explore
By company size
By use case
By industry
View all solutions
Topics
- AI
- DevOps
- Security
- Software Development
- View all
Explore
- GitHub Sponsors
  Fund open source developers
- The ReadME Project
  GitHub community articles
Repositories
- Enterprise platform
  AI-powered developer platform
Available add-ons
Pricing

Search code, repositories, users, issues, pull requests...

Search

Clear

Search syntax tips

Provide feedback

We read every piece of feedback, and take your input very seriously.

Include my email address so I can be contacted

Saved searches

Use saved searches to filter your results more quickly

Name

Query

To see all available qualifiers, see our documentation.

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

NVIDIA / TransformerEngine Public

Notifications You must be signed in to change notification settings
Fork 337
Star 2k

Code
Issues 157
Pull requests 44
Discussions
Actions
Security
Insights

Additional navigation options

Code
Issues
Pull requests
Discussions
Actions
Security
Insights

[C/PyTorch] Add support for multi-latent attention (MLA)#1039

Merged

cyanguwa merged 19 commits intoNVIDIA:mainNVIDIA/TransformerEngine:mainfrom cyanguwa:add_mlacyanguwa/TransformerEngine:add_mlaCopy head branch name to clipboard

Aug 6, 2024

+343-250

Conversation Commits19 (19)Checks Files changed

Commits

Commits on Jul 24, 2024

add multi-latent attention for DPA
cyanguwa
committed
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed
fix Jax/Paddle API
cyanguwa
committed
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed
fix lint
cyanguwa
committed

Commits on Jul 25, 2024

fix typo in test script
cyanguwa
committed
fix too-many-boolean lint error
cyanguwa
committed
Revert "fix lint"
cyanguwa
committed
Merge branch 'main' into add_mla
cyanguwa
authored
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed

Commits on Jul 30, 2024

fix stride check in get_qkv_layout
cyanguwa
committed
Merge branch 'NVIDIA:main' into add_mla
cyanguwa
authored
WIP: fix layout_thd tests
cyanguwa
committed
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed
WIP: debug info
cyanguwa
committed
fix merge conflict
cyanguwa
committed
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed
fix thd pad_between_seqs=False/True tests
cyanguwa
committed
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed

Footer

© 2024 GitHub, Inc.

Footer navigation

Terms
Privacy
Security
Status
Docs
Contact

You can’t perform that action at this time.