Skip to content

Pull requests: pytorch/ao

New pull request New
223 Open 2,750 Closed
223 Open 2,750 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

docs: Add FqnToConfig to API reference quantization page CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3709 opened Jan 23, 2026 by pantha704 Loading…
1
[mxfp8 moe training] fallback cuda kernel for when input doesn't meet 2d TMA constraints 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: new feature Use this tag if this PR adds a new feature
#3708 opened Jan 23, 2026 by danielvegamyhre Loading…
1
Update TorchAO API usage to V2 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
#3707 opened Jan 23, 2026 by RandySheriff Loading…
2
Added benchmarking on SD3.5 for new fp8 sdpa API CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3706 opened Jan 23, 2026 by howardzhang-cv Loading…
1
refactor integer static quantization flow CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. enhancement New feature or request
#3687 opened Jan 22, 2026 by namgyu-youn Loading…
4
[mx] Fix: pass scaling_mode parameter to triton_to_mxfp8_dim1 CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3686 opened Jan 22, 2026 by ugolowic Loading…
1
Fix zero-point handling for packed_weight_inputs in INT8 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
#3685 opened Jan 22, 2026 by Yuxingwang-intel Loading…
3
[ROCm] Enable Float8 base test suite CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: rocm
#3682 opened Jan 22, 2026 by ZhiweiYan-96 Loading…
8
Consolidate enum for observer step 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
#3678 opened Jan 21, 2026 by namgyu-youn Loading…
12
make torchao import not initialize cuda CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: bug fix Use this tag for PRs that fix bugs
#3676 opened Jan 21, 2026 by vkuzo Loading…
1
3
[ROCM] Bring back Float8 test integrations unit tests CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: rocm
#3674 opened Jan 21, 2026 by alex-minooka Loading…
1
[ROCM] Float8 Test Base Unit Tests for 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
#3673 opened Jan 21, 2026 by alex-minooka Loading…
1
[ROCM] Bringing Float8 Util Unit Tests for 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
#3672 opened Jan 21, 2026 by alex-minooka Loading…
1
[mxfp8 moe training] updates to mxfp8 ep bench script CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3671 opened Jan 21, 2026 by danielvegamyhre • Draft
1
add mxfp8 and nvfp4 support to safetensors 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)
#3668 opened Jan 21, 2026 by liangel-02 Loading…
4
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] 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
(prototype) add support for MXFP8 and MXFP4 QAT 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
#3644 opened Jan 15, 2026 by ved1beta Loading…
49
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…
2
[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…
8
[xpu][test] Remove None in get_current_accelerator_device ciflow/xpu label used to trigger xpu CI jobs 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) xpu Intel XPU related features
#3634 opened Jan 14, 2026 by zxd1997066 Loading…
14
[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
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…
13
ProTip! no:milestone will show everything without a milestone.