Skip to content

Pull requests: pytorch/ao

New pull request New
315 Open 3,180 Closed
315 Open 3,180 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

[PT2E] Run weight observer eagerly for dynamic quant CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: not user facing Use this tag if you don't want this PR to show up in release notes module: pt2e_quant pt2 export quantization (prepare_pt2e, convert_pt2e, quantizer)
#4307 opened Apr 21, 2026 by Xia-Weiwen Collaborator • Draft
2
[mxfp8 moe training] integrate 1x32 cutedsl quantization kernel CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4306 opened Apr 21, 2026 by danielvegamyhre Contributor Loading…
1
docs(qat): update supported configs list, fix unclickable link, add blog ref
#4305 opened Apr 21, 2026 by Anai-Guo Loading…
1
2
[ROCm] test(moe): skip jagged rowwise scales test. 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
#4304 opened Apr 20, 2026 by jagadish-amd Loading…
3
[wip] gptq + nvfp4 e2e test CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: not user facing Use this tag if you don't want this PR to show up in release notes
#4303 opened Apr 20, 2026 by vkuzo Contributor Loading…
2
hook up gptq prototype to nvfp4 CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: not user facing Use this tag if you don't want this PR to show up in release notes
#4302 opened Apr 20, 2026 by vkuzo Contributor Loading…
10
[X86] Bug fixes and refinements in da8w4/float8 linear kernels CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. cpu module: not user facing Use this tag if you don't want this PR to show up in release notes
#4301 opened Apr 20, 2026 by Xia-Weiwen Collaborator Loading…
7
docs: fix Int8Tensor row in Tensor Subclasses list-table CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4299 opened Apr 20, 2026 by MukundaKatta Loading…
1
[test][xpu] Adjust the test sequence for Intel GPU CI CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4298 opened Apr 19, 2026 by zxd1997066 Contributor Loading…
1
fix(utils): propagate non_blocking in TorchAOBaseTensor._to_copy and _get_to_kwargs CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4297 opened Apr 19, 2026 by Dev-next-gen Loading…
4
small fix to low precision attention benchmarking to force baseline contexts CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: not user facing Use this tag if you don't want this PR to show up in release notes
#4296 opened Apr 18, 2026 by howardzhang-cv Contributor Loading…
1
add FSDP and TP tests for Float8BlockwiseLinear CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: training quantize_ api training flow
#4295 opened Apr 17, 2026 by iamzainhuda Contributor Loading…
2
Move collect_producer_nodes to graph_utils.py CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4294 opened Apr 17, 2026 by tom-arm Contributor Loading…
1
[mxfp8 moe training] fuse dynamic per-group padding into cutedsl 2d mxfp8 quant kernel CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4293 opened Apr 17, 2026 by MagellaX Contributor Loading…
1
6
basic enablement for mxfp8 and mxfp4 inference on AMD MI350x CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: inference quantize_ api inference flow
#4290 opened Apr 16, 2026 by vkuzo Contributor Loading…
4
Add iterative reweighting to PruneOptimizer CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: not user facing Use this tag if you don't want this PR to show up in release notes
#4283 opened Apr 15, 2026 by lisjin Contributor Loading…
6
[AARCH64] Enable MKLDNN Backend for Int8DynamicActivationInt8WeightConfig() on ARM CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4281 opened Apr 15, 2026 by agrawal-aka Contributor Loading…
1
Add Sparse2x4HIPSPARSELTFloat8Tensor (#4277) 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 module: rocm
#4277 opened Apr 14, 2026 by bbeckca Contributor Loading…
6
[ROCm] Add MXFP8 training support for gfx950 (MI355X) 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
#4275 opened Apr 14, 2026 by indianspeedster Loading… MXFP8 Training
13
[xpu][mx][test] Enable mx serialization tests on xpu CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4272 opened Apr 13, 2026 by ugolowic Contributor Loading…
1
[xpu][mx] Fix NaN scale propagation in RCEIL triton kernel CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4271 opened Apr 13, 2026 by ugolowic Contributor Loading…
3
[optim] Add GrokAdamW optimizer with low-bit quantization support CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4270 opened Apr 13, 2026 by vaibhavhariram Loading…
1
3
Add torch.uint16, torch.uint32 CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4269 opened Apr 12, 2026 by Freed-Wu Loading…
1
Add reduce_range to avoid overflow in int8 tensor CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: not user facing Use this tag if you don't want this PR to show up in release notes
#4266 opened Apr 10, 2026 by cyxlily Contributor Loading…
29
Support 32x32 scaling for weights in MXFP8 weight quantization kernel CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: training quantize_ api training flow moe mx
#4254 opened Apr 9, 2026 by alexsamardzic Collaborator Loading…
1
21
ProTip! Exclude everything labeled bug with -label:bug.