Skip to content

Pull requests: pytorch/ao

New pull request New
257 Open 2,952 Closed
257 Open 2,952 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 wrong type annotations CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3974 opened Mar 2, 2026 by Freed-Wu Loading…
1
Add fused FP8 rowwise scale+cast kernel for MoE forward pass 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
#3973 opened Mar 2, 2026 by lizamd Loading…
2 of 3 tasks
2
Optimize FP8 colwise scales kernel for AMD GPUs in MoE backward pass CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. moe mx
#3972 opened Mar 2, 2026 by lizamd Loading…
3 of 4 tasks
5
Support conv_transpose2d CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3971 opened Mar 1, 2026 by Freed-Wu Loading…
1
Support kwargs for conv CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3970 opened Mar 1, 2026 by Freed-Wu Loading…
1
Support PerTensor for IntxWeightOnlyConfig CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3969 opened Mar 1, 2026 by Freed-Wu Loading…
1
[WIP] unified tensor subclass for training CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3968 opened Feb 28, 2026 by danielvegamyhre Loading…
1
fix: fix a typo CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3962 opened Feb 27, 2026 by Freed-Wu Loading…
1
Add pre-quantized activation support to MXFP8 grouped GEMM (_to_mxfp8_then_scaled_grouped_mm) 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)
#3961 opened Feb 27, 2026 by MagellaX Loading…
8
Add FA4 monkey-patch path for low-precision attention 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
#3960 opened Feb 27, 2026 by howardzhang-cv • Draft
2
Add FP8 FA3 low-precision attention with monkey-patch SDPA path 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
#3959 opened Feb 27, 2026 by howardzhang-cv Loading…
1
Patch nightly Linux AArch64-specific wheel build CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3954 opened Feb 26, 2026 by puneetmatharu Loading…
2
Fix float8 int4 transform to use parameter_name instead of hardcoded module.weight CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: qat quantize_ api qat flow topic: bug fix Use this tag for PRs that fix bugs
#3953 opened Feb 26, 2026 by petrex Loading…
3 tasks done
1
Fix operator precedence bug in is_Navi4() GPU detection CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3951 opened Feb 25, 2026 by brucechanglongxu Loading…
3 tasks done
2
Add better debug print for failed prepare 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
#3950 opened Feb 25, 2026 by JakeStevens Loading…
2
[mxfp8 training] remove mxfp8 from MXLinear and MXLinearConfig 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 mx
#3949 opened Feb 25, 2026 by danielvegamyhre Loading…
1
[mxfp8 training] unified MXFP8TrainingConfig and MXFP8TrainingTensor 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
#3948 opened Feb 25, 2026 by danielvegamyhre Loading…
1
Add FA4 fp8 backend to low precision attention api 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
#3947 opened Feb 25, 2026 by howardzhang-cv • Draft
1
Add RunningAbsMaxSmoothQuantObserver for memory-efficient calibration 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
#3946 opened Feb 25, 2026 by jcaip Loading…
4
[ROCm] Enable float8 utils and base tests on 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
#3942 opened Feb 24, 2026 by brucechanglongxu Loading…
2
[ROCm] Enable affine quantized, quant API, and integration tests on 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
#3941 opened Feb 24, 2026 by brucechanglongxu Loading…
2
[ROCm] Remove redundant ROCm test skips in NF4 and use accelerator-ag… CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: rocm
#3940 opened Feb 24, 2026 by brucechanglongxu Loading…
2
[ROCm] Enable low-bit optimizer tests on 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
#3939 opened Feb 24, 2026 by brucechanglongxu Loading…
3
lowbit linear weight packing and shared embedding on x86 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
#3938 opened Feb 24, 2026 by JacobSzwejbka Loading…
2
Add MXFP4 support for GPTQ quantization and separate eval script CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3935 opened Feb 23, 2026 by jcaip Loading…
1
ProTip! Updated in the last three days: updated:>2026-02-27.