Skip to content
Navigation Menu
Sign in
Appearance settings
Platform
AI CODE CREATION
GitHub Copilot
Write better code with AI
GitHub Copilot app
Direct agents from issue to merge
MCP Registry
Integrate external tools
DEVELOPER WORKFLOWS
Actions
Automate any workflow
Codespaces
Instant dev environments
Issues
Plan and track work
Code Review
Manage code changes
Code Quality
Enforce quality at merge
APPLICATION SECURITY
GitHub Advanced Security
Find and fix vulnerabilities
Code security
Secure your code as you build
Secret protection
Stop leaks before they start
EXPLORE
Why GitHub
Documentation
Blog
Changelog
Marketplace
View all features
Solutions
BY COMPANY SIZE
Enterprises
Small and medium teams
Startups
Nonprofits
BY USE CASE
App Modernization
DevSecOps
DevOps
CI/CD
View all use cases
BY INDUSTRY
Healthcare
Financial services
Manufacturing
Government
View all industries
View all solutions
Resources
EXPLORE BY TOPIC
AI
Software Development
DevOps
Security
View all topics
EXPLORE BY TYPE
Customer stories
Events & webinars
Ebooks & reports
Business insights
GitHub Skills
SUPPORT & SERVICES
Documentation
Customer support
Community forum
Trust center
Partners
View all resources
Open Source
COMMUNITY
GitHub Sponsors
Fund open source developers
PROGRAMS
Security Lab
Maintainer Community
Accelerator
GitHub Stars
Archive Program
REPOSITORIES
Topics
Trending
Collections
Enterprise
ENTERPRISE SOLUTIONS
Enterprise platform
AI-powered developer platform
AVAILABLE ADD-ONS
GitHub Advanced Security
Enterprise-grade security features
Copilot for Business
Enterprise-grade AI features
Premium Support
Enterprise-grade 24/7 support
Pricing
Type
/
to search
Sign in
Sign up
Appearance settings
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
{{ message }}
Uh oh!
There was an error while loading.
Please reload this page
.
NVIDIA
/
TransformerEngine
Public
Notifications
You must be signed in to change notification settings
Fork
801
Star
3.5k
Code
Issues
161
Pull requests
165
Discussions
Actions
Security and quality
0
Insights
Additional navigation options
Code
Issues
Pull requests
Discussions
Actions
Security and quality
Insights
[PyTorch] Optimize MXFP8 all-gathers
- #1581
#1581
Merged
timmoon10
merged 11 commits into
NVIDIA:main
NVIDIA/TransformerEngine:main
from
timmoon10:optimize-wgrad-allgather
timmoon10/TransformerEngine:optimize-wgrad-allgather
Copy head branch name to clipboard
Mar 25, 2025
Conversation
Commits
11
(11)
Checks
Files changed
Merged
[PyTorch] Optimize MXFP8 all-gathers
#1581
timmoon10
merged 11 commits into
NVIDIA:main
NVIDIA/TransformerEngine:main
from
timmoon10:optimize-wgrad-allgather
timmoon10/TransformerEngine:optimize-wgrad-allgather
Copy head branch name to clipboard
Commits
Commits on Mar 14, 2025
Coalesce NCCL all-gathers for MXFP8 all-gather
Show description for 0c1a98f
timmoon10
committed
0c1a98f
View commit details
Copy full SHA for 0c1a98f
Browse repository at this point
Commits on Mar 15, 2025
Merge branch 'main' into optimize-wgrad-allgather
timmoon10
committed
15c34ec
View commit details
Copy full SHA for 15c34ec
Browse repository at this point
Add missing import
Show description for 79e0dad
timmoon10
committed
79e0dad
View commit details
Copy full SHA for 79e0dad
Browse repository at this point
[pre-commit.ci] auto fixes from pre-commit.com hooks
Show description for 32763b1
pre-commit-ci[bot]
committed
32763b1
View commit details
Copy full SHA for 32763b1
Browse repository at this point
Commits on Mar 22, 2025
Merge branch 'main' into optimize-wgrad-allgather
timmoon10
committed
59290e1
View commit details
Copy full SHA for 59290e1
Browse repository at this point
Merge branch 'main' into optimize-wgrad-allgather
timmoon10
committed
5c13e12
View commit details
Copy full SHA for 5c13e12
Browse repository at this point
Cache quantized input tensor after linear module forward pass
Show description for de2f7de
timmoon10
committed
de2f7de
View commit details
Copy full SHA for de2f7de
Browse repository at this point
[pre-commit.ci] auto fixes from pre-commit.com hooks
Show description for 38a4c09
pre-commit-ci[bot]
committed
38a4c09
View commit details
Copy full SHA for 38a4c09
Browse repository at this point
Fix linter warnings
Show description for 9acd9cf
timmoon10
committed
9acd9cf
View commit details
Copy full SHA for 9acd9cf
Browse repository at this point
Commits on Mar 25, 2025
Avoid unnecessarily allocating layernorm output in LayerNormLinear/LayerNormMLP
Show description for 9e3f392
timmoon10
committed
9e3f392
View commit details
Copy full SHA for 9e3f392
Browse repository at this point
Merge branch 'main' into optimize-wgrad-allgather
timmoon10
committed
baa03c1
View commit details
Copy full SHA for baa03c1
Browse repository at this point
You can’t perform that action at this time.
![pre-commit-ci[bot]](https://avatars.githubusercontent.com/in/68672?v=4&size=32){kind=avatar}