Skip to content

Pull requests: pytorch/ao

New pull request New
235 Open 2,792 Closed
235 Open 2,792 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

Fix rowwise_scaled_linear_sparse_cutlass test 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
#3768 opened Jan 29, 2026 by andrewor14 Loading…
3
[9/x] docs update: remove empty kernel section 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
#3767 opened Jan 29, 2026 by vkuzo Loading…
2
[8/x] docs update: fix KernelPreference docstring 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
#3766 opened Jan 29, 2026 by vkuzo Loading…
2
[7/x] docs update: float8 training api reference 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
#3765 opened Jan 29, 2026 by vkuzo Loading…
2
Update GPTQ to use common enum step 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
#3757 opened Jan 29, 2026 by namgyu-youn Loading…
2
Support FP8 output for scaled_embedding_bag 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
#3755 opened Jan 29, 2026 by yanbing-j Loading…
9
[mxfp8 moe training] fix bug in bench_ep_pipeline script 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
#3753 opened Jan 29, 2026 by danielvegamyhre Loading…
1
[mxfp8 moe training][docs] add tutorial for training with MXFP8 expert parallel 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: documentation Use this tag if this PR adds or improves documentation
#3752 opened Jan 29, 2026 by danielvegamyhre Loading…
2
[mxfp8 moe training] add MXFP8 expert parallel example 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: documentation Use this tag if this PR adds or improves documentation
#3751 opened Jan 29, 2026 by danielvegamyhre Loading…
1
[mxfp8 moe training] remove outdated example 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: documentation Use this tag if this PR adds or improves documentation
#3750 opened Jan 29, 2026 by danielvegamyhre Loading…
1
[metal] Enable bias in metal lowbit-quantized linear kernels 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
#3745 opened Jan 28, 2026 by manuelcandales Loading…
1
[DRAFT] Move all torch._scaled_mm calls to F.scaled_mm 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
#3742 opened Jan 28, 2026 by slayton58 Loading…
17
[mxfp8 moe training] remove use_cuda_kernel_for_blocked_layout param 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: bc-breaking Use this tag if this PR breaks backward compatibility
#3735 opened Jan 27, 2026 by danielvegamyhre Loading…
1
[mxfp8 moe training] remove "use_triton_for_dim0_cast" flag, infer dim0 quant kernel from kernel_preference 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: bc-breaking Use this tag if this PR breaks backward compatibility
#3734 opened Jan 27, 2026 by danielvegamyhre Loading…
1
Added new API to fp8_rope_sdpa_inference: RoPE+Hadamard+fp8 quantization kernel into fp8 sdpa 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
#3732 opened Jan 27, 2026 by howardzhang-cv • Draft
1
Added new API to benchmark file: RoPE+fp8 quantization kernel into fp8 SDPA 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
#3731 opened Jan 27, 2026 by howardzhang-cv Loading…
1
Added new fp8_rope_sdpa_inference API 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
#3730 opened Jan 27, 2026 by howardzhang-cv Loading…
7
fp8_sdpa torch.compile functionality 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
#3729 opened Jan 27, 2026 by howardzhang-cv Loading…
2
Make to_sparse_semi_structured_cutlass_sm9x 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: not user facing Use this tag if you don't want this PR to show up in release notes
#3727 opened Jan 26, 2026 by jerryzh168 Loading…
20
[claude] Make rowwise_scaled_linear_sparse_cutlass 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)
#3725 opened Jan 26, 2026 by andrewor14 Loading…
17
[mxfp8 moe training] add emulated mode 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
#3724 opened Jan 26, 2026 by danielvegamyhre Loading…
10
Add reference link for MSLK kernel 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
#3717 opened Jan 25, 2026 by namgyu-youn Loading…
2
Implement aten.abs.default for Sparse2x4CUTLASSFloat8Tensor 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
#3716 opened Jan 24, 2026 by RandySheriff Loading…
5
Add plain int4_packing_format support for Float8DynamicActivationInt4WeightConfig 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
#3714 opened Jan 23, 2026 by jerryzh168 Loading…
4
fix dynamo trace issue and bug introduced by new float8_dtype param 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
#3712 opened Jan 23, 2026 by danielvegamyhre • Draft
1
ProTip! Updated in the last three days: updated:>2026-01-26.