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 }}
MagellaX
/
ao
Public
forked from
pytorch/ao
Notifications
You must be signed in to change notification settings
Fork
0
Star
0
Code
Pull requests
0
Actions
Projects
Security and quality
0
Insights
Additional navigation options
Code
Pull requests
Actions
Projects
Security and quality
Insights
Commits
Branch selector
main
User selector
All users
All time
Commit history
Commits on Nov 4, 2025
[mxfp8 moe training] compute prefix sum of group sizes inside kernel intead of precomputing (#3285)
danielvegamyhre
authored
01374eb
View commit details
Copy full SHA for 01374eb
Browse repository at this point
fix unit test to use no grad (#3283)
liangel-02
authored
8e3b3da
View commit details
Copy full SHA for 8e3b3da
Browse repository at this point
updating flatten/unflatten functions (#3282)
liangel-02
authored
ba3ac9f
View commit details
Copy full SHA for ba3ac9f
Browse repository at this point
Commits on Nov 3, 2025
Use nn_module_stack instead
Show description for 9266734
tugsbayasgalan
authored
9266734
View commit details
Copy full SHA for 9266734
Browse repository at this point
[mxfp8 moe training] make scaling mode configurable and make rceil default (#3271)
danielvegamyhre
authored
f657903
View commit details
Copy full SHA for f657903
Browse repository at this point
Set model parameters to requires_grad=False (#3272)
Show description for 2a89491
Jesse Cai
authored
2a89491
View commit details
Copy full SHA for 2a89491
Browse repository at this point
Commits on Oct 31, 2025
[mxfp8] fix test nan != nan issue (#3273)
danielvegamyhre
authored
315e9b4
View commit details
Copy full SHA for 315e9b4
Browse repository at this point
add bias handling for a_1_128_w_128_128 float8 scaling (#3259)
Show description for f856d36
vkuzo
authored
f856d36
View commit details
Copy full SHA for f856d36
Browse repository at this point
add a_1_128_w_128_128 (DeepSeek) float8 scaling for inference (#3257)
Show description for b49178c
vkuzo
authored
b49178c
View commit details
Copy full SHA for b49178c
Browse repository at this point
Update torchao + unsloth integration on README (#3267)
Show description for e418066
andrewor14
authored
e418066
View commit details
Copy full SHA for e418066
Browse repository at this point
Add per tensor fp8 quantization support for conv3d (#3215)
Show description for 258387a
jerryzh168
authored
258387a
View commit details
Copy full SHA for 258387a
Browse repository at this point
Commits on Oct 30, 2025
move float8 blockwise kernels out of prototype (#3256)
Show description for 1e473ed
vkuzo
authored
1e473ed
View commit details
Copy full SHA for 1e473ed
Browse repository at this point
properly skip float8 inference tests without fbgemm (#3255)
Show description for d3bec87
vkuzo
authored
d3bec87
View commit details
Copy full SHA for d3bec87
Browse repository at this point
[mxfp8 moe training] update readme and tests (#3260)
danielvegamyhre
authored
0f05b40
View commit details
Copy full SHA for 0f05b40
Browse repository at this point
Commits on Oct 29, 2025
[CPU] Improve INT8 SDPA template (#3230)
Show description for 3577306
Xia-Weiwen
authored
3577306
View commit details
Copy full SHA for 3577306
Browse repository at this point
[mxfp8 moe training] initialize zero tensor differently to avoid d2h sync (#3253)
danielvegamyhre
authored
d84f5b8
View commit details
Copy full SHA for d84f5b8
Browse repository at this point
Commits on Oct 28, 2025
Add quantize_ nn.Parameter support (#3083)
Show description for 50a555b
Jesse Cai
authored
50a555b
View commit details
Copy full SHA for 50a555b
Browse repository at this point
[mxfp8 moe training] bench and profile mxfp8 a2a fwd and bwd separately (#3203)
danielvegamyhre
authored
dffb3a0
View commit details
Copy full SHA for dffb3a0
Browse repository at this point
[mxfp8 moe training] simplify e8m0 -> fp32 calc (#3201)
danielvegamyhre
authored
d089c6a
View commit details
Copy full SHA for d089c6a
Browse repository at this point
[mxfp8 moe training] integrate triton quant/dequant kernels into mxfp8 all to all (#3197)
danielvegamyhre
authored
6a62549
View commit details
Copy full SHA for 6a62549
Browse repository at this point
[mxfp8 moe training] add triton kernel for mxfp8 dequantization (#3195)
danielvegamyhre
authored
7537d99
View commit details
Copy full SHA for 7537d99
Browse repository at this point
create a separate test for mx and nv serialization (#3251)
Show description for 41a0778
vkuzo
authored
41a0778
View commit details
Copy full SHA for 41a0778
Browse repository at this point
Commits on Oct 27, 2025
nvfp4: support inference_mode and rank 3 (#3240)
Show description for 03c2d28
vkuzo
authored
03c2d28
View commit details
Copy full SHA for 03c2d28
Browse repository at this point
mx: support inference_mode and rank 3+ (#3238)
Show description for 14eff10
vkuzo
authored
14eff10
View commit details
Copy full SHA for 14eff10
Browse repository at this point
Commits on Oct 25, 2025
Only convert to int4 preshuffled tensor in H100 (#3245)
Show description for f303f4c
jerryzh168
authored
f303f4c
View commit details
Copy full SHA for f303f4c
Browse repository at this point
Commits on Oct 24, 2025
[moe training] update readme with links, cleanup (#3239)
danielvegamyhre
authored
e9c7bea
View commit details
Copy full SHA for e9c7bea
Browse repository at this point
Commits on Oct 23, 2025
Fix Wq size check
Show description for 53b5efd
RandySheriff
authored
53b5efd
View commit details
Copy full SHA for 53b5efd
Browse repository at this point
docs: fix qat description in README.md (#3212)
orangeH25
authored
204cd48
View commit details
Copy full SHA for 204cd48
Browse repository at this point
Add Unsloth + QAT blog to latest news (#3227)
andrewor14
authored
f3fc5e7
View commit details
Copy full SHA for f3fc5e7
Browse repository at this point
Commits on Oct 22, 2025
Update README.md (#3225)
Show description for 54a7b85
jerryzh168
authored
54a7b85
View commit details
Copy full SHA for 54a7b85
Browse repository at this point
[mxfp8 moe training] make compile vs triton for dim0 cast configurable (#3219)
danielvegamyhre
authored
7e68d5e
View commit details
Copy full SHA for 7e68d5e
Browse repository at this point
Commits on Oct 21, 2025
enable custom MKN in inference roofline script (#3224)
Show description for beee153
vkuzo
authored
beee153
View commit details
Copy full SHA for beee153
Browse repository at this point
mxfp8 inference roofline: add fusion to observed (#3223)
Show description for 6452b4a
vkuzo
authored
6452b4a
View commit details
Copy full SHA for 6452b4a
Browse repository at this point
fix mxfp8 matmul benchmark (#3221)
Show description for 13595c5
vkuzo
authored
13595c5
View commit details
Copy full SHA for 13595c5
Browse repository at this point
[moe training] change api _scaled_grouped_mm -> _quantize_then_scaled_grouped_mm (#3218)
danielvegamyhre
authored
82ae011
View commit details
Copy full SHA for 82ae011
Browse repository at this point
Previous
Next
You can’t perform that action at this time.