Skip to content

Pull requests: pytorch/ao

New pull request New
196 Open 2,568 Closed
196 Open 2,568 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

use the float4 dtype in mxfp4 and nvfp4 tensors 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
#3440 opened Dec 4, 2025 by vkuzo Loading…
2
Update shapes in microbenchmarks to reflect more realistic data ciflow/benchmark 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
#3438 opened Dec 4, 2025 by jainapurva • Draft
1
Add support for MXFP8 All gather CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3435 opened Dec 4, 2025 by avizon-aws Loading…
7
[reland][xpu][test] Port 2 test/prototype/test_{parq, quantized_training} UT files to intel XPU ci-no-td CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3434 opened Dec 4, 2025 by zxd1997066 Loading…
2
[CPU] Reland qconv fp8 fusion passes ci-no-td ciflow/rocm 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
#3433 opened Dec 4, 2025 by Xia-Weiwen Loading…
1
[WIP] Merge all model eval scripts together CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3430 opened Dec 3, 2025 by jainapurva • Draft
1
Add pruning-aware training in torchao.prototype.pat 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
#3429 opened Dec 3, 2025 by lisjin Loading…
1
not for land: vllm bench for mxfp8 CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3426 opened Dec 3, 2025 by vkuzo Loading…
2
[xpu][test] Port test_float8_utils.py and test_hqq_affine.py UT files to intel XPU CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3424 opened Dec 3, 2025 by zxd1997066 Loading…
2
[xpu][test] Port 2 test/prototype/test_{nvfp4_tensor, spinquant} UT files to intel XPU CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3423 opened Dec 3, 2025 by zxd1997066 Loading…
2
Static smooth quant CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3422 opened Dec 3, 2025 by cyxlily Loading…
1
Roofline quantized conv3d/2d layer CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3419 opened Dec 3, 2025 by jainapurva • Draft
7
[mxfp8 moe training] parallelize along col blocks in scale blocked format kernel for groups along K 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: not user facing Use this tag if you don't want this PR to show up in release notes
#3416 opened Dec 2, 2025 by danielvegamyhre • Draft
1
[xpu][test] Port 2 test/prototype/mx_formats/test_{kernels, test_mx_tensor} UT files to intel XPU 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: for developers Use this tag if this PR is mainly developer facing xpu Intel XPU related features
#3409 opened Dec 2, 2025 by zxd1997066 Loading…
3
Int8Tensor migration cleanup 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)
#3407 opened Dec 1, 2025 by jcaip Loading…
36
support fp8 scaled_embedding_bag pattern match ciflow/rocm 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
#3406 opened Dec 1, 2025 by shiyang-weng • Draft
7
[xpu][test] Port 2 test/quantization/pt2e/test_{quantize_pt2e, quantize_pt2e_qat} UT files to intel XPU 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 xpu Intel XPU related features
#3405 opened Nov 30, 2025 by zxd1997066 Loading…
4
Introduce Stochastic Rounding for low-bit training 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
#3404 opened Nov 30, 2025 by namgyu-youn Loading…
2
Build PERF for HQQ vs. Affine CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: performance Use this tag if this PR improves the performance of a feature
#3403 opened Nov 30, 2025 by namgyu-youn Loading…
8
Fixing subclassing CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3401 opened Nov 27, 2025 by Krishn1412 Loading…
2
[xpu][test] Port test_integration.py to intel XPU 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: for developers Use this tag if this PR is mainly developer facing xpu Intel XPU related features
#3400 opened Nov 27, 2025 by zxd1997066 Loading…
2
[WIP] Implements rounding mode for NVFP4 tensor CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3384 opened Nov 24, 2025 by syed-ahmed • Draft
1
1
Davided0/feedforward horizontal fusion CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3380 opened Nov 24, 2025 by davided0 Loading…
1
1
[xpu][test] Port 2 test/float8_{base, compile} UT files to intel XPU 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: for developers Use this tag if this PR is mainly developer facing xpu Intel XPU related features
#3377 opened Nov 24, 2025 by zxd1997066 Loading…
6
[WIP] update keepdim params for 1D scale factor 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
#3374 opened Nov 22, 2025 by namgyu-youn • Draft
4
ProTip! Mix and match filters to narrow down what you’re looking for.