Skip to content

Pull requests: pytorch/ao

New pull request New
221 Open 2,722 Closed
221 Open 2,722 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

[mxfp8 moe training] auto-select chunk_width in cuda blocked layout kernel CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. moe mx topic: improvement Use this tag if this PR is an improvement (doesn't fit into any of the other categories)
#3658 opened Jan 17, 2026 by danielvegamyhre Loading…
1
fix: avoid attribute assignment on typing.Union for Python 3.14 compatibility in pt2e CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3657 opened Jan 17, 2026 by lordaarush Loading…
4 tasks done
1
1
[mxfp8 moe training] cuda blocked layout kernel handling for skinnier scale tensors CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. moe mx topic: bug fix Use this tag for PRs that fix bugs topic: improvement Use this tag if this PR is an improvement (doesn't fit into any of the other categories)
#3656 opened Jan 17, 2026 by danielvegamyhre Loading…
1
1
Add static quant flow support for Float8StaticActivationFloat8WeightConfig CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: not user facing Use this tag if you don't want this PR to show up in release notes
#3655 opened Jan 17, 2026 by jerryzh168 Loading…
6
Deprecate v1 of Float8StaticActivationFloat8WeightConfig CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: not user facing Use this tag if you don't want this PR to show up in release notes
#3654 opened Jan 17, 2026 by jerryzh168 Loading…
2
Update TorchAOBaseTensor docs CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: documentation Use this tag if this PR adds or improves documentation
#3652 opened Jan 17, 2026 by jerryzh168 Loading…
4
[mxfp8 moe training] rename MoEScalingType -> ScaledGroupedMMRecipe CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. moe mx topic: improvement Use this tag if this PR is an improvement (doesn't fit into any of the other categories)
#3651 opened Jan 16, 2026 by danielvegamyhre Loading…
3
Fix: proper granularity check CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3649 opened Jan 16, 2026 by S1ro1 Loading…
1
mvfp4 support ao CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3644 opened Jan 15, 2026 by ved1beta Loading…
14
[ROCM] Fix float8 rounding test case to enable ROCM CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: rocm
#3643 opened Jan 15, 2026 by alex-minooka Loading…
8
Allow non-tensor kwargs in prepare_pt2e CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3642 opened Jan 15, 2026 by Erik-Lundell Loading…
12
Added a prototype folder for experimenting with fp8 sdpa inference CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: not user facing Use this tag if you don't want this PR to show up in release notes
#3641 opened Jan 15, 2026 by howardzhang-cv Loading…
1
[DO NOT MERGE] Test changes to linux_job_v2.yml via regression_test_rocm.yml ciflow/rocm ciflow/rocm-mi300 CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: rocm
#3638 opened Jan 14, 2026 by jithunnair-amd • Draft
2
Unify float8 scale selection for per-tensor quantization CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3635 opened Jan 14, 2026 by MagellaX Loading…
5
[xpu][test] Remove None in get_current_accelerator_device CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3634 opened Jan 14, 2026 by zxd1997066 Loading…
2
[ROCm] bring some mxfp8 quantization unit test back CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: rocm
#3628 opened Jan 13, 2026 by RuibinCheung Loading…
5
[WIP] Upgrade AO to MSLK CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported meta-exported topic: not user facing Use this tag if you don't want this PR to show up in release notes
#3622 opened Jan 12, 2026 by cthi Loading…
11
Deprecate TransformerEvalWrapper, LMEvalInputRecorder CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: deprecation Use this tag if this PR deprecates a feature
#3617 opened Jan 10, 2026 by namgyu-youn Loading…
12
[not for land] profiling instrumentation for mx blocked layout kernel CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3616 opened Jan 10, 2026 by danielvegamyhre Loading…
1
Add aten.narrow support for inference_mode in Float8Tensor CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: new feature Use this tag if this PR adds a new feature
#3615 opened Jan 10, 2026 by jerryzh168 Loading…
3 tasks done
2
[claude] Make MXFP8 cuda kernels ABI stable CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: improvement Use this tag if this PR is an improvement (doesn't fit into any of the other categories)
#3610 opened Jan 9, 2026 by andrewor14 Loading…
16
Add AWQ/SmoothQuant to benchmark module CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: for developers Use this tag if this PR is mainly developer facing
#3602 opened Jan 8, 2026 by namgyu-youn Loading…
1
21
Update Int8WeightOnlyConfig and Int8DynamicActivationInt8WeightConfig to use version=2 by default CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: bc-breaking Use this tag if this PR breaks backward compatibility topic: documentation Use this tag if this PR adds or improves documentation
#3597 opened Jan 7, 2026 by jcaip Loading…
3
Deprecate v1 of W8A16-INT, W8A8-INT quantization subclass CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: deprecation Use this tag if this PR deprecates a feature
#3592 opened Jan 7, 2026 by namgyu-youn Loading…
2
Remove external wrapper in SmoothQuant CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: improvement Use this tag if this PR is an improvement (doesn't fit into any of the other categories)
#3591 opened Jan 7, 2026 by namgyu-youn Loading…
7
ProTip! Type g i on any issue or pull request to go back to the issue listing page.