Search results

[Feature request] Make offset functions part of ukernel configs in torchao/experimental

cpu

enhancementNew feature or requestNew feature or request

Status: Open.

#1745 In pytorch/ao;

· metascroy opened on Feb 20, 2025

[Feature request] Use guard TORCHAO_ENABLE_ARM_DOTPROD in torchao code

build

enhancementNew feature or requestNew feature or request

Status: Open.

#1743 In pytorch/ao;

· metascroy opened on Feb 20, 2025

Sparsity training together with FP8 linear layer in torchao

enhancementNew feature or requestNew feature or request

sparsity

Status: Open.

#1574 In pytorch/ao;

· goldhuang opened on Jan 17, 2025

fuse ConvTranspose2d+BatchNorm2d+ReLU

enhancementNew feature or requestNew feature or request

Status: Open.

#1462 In pytorch/ao;

· yokosyun opened on Dec 28, 2024

Low bit shampoo

enhancementNew feature or requestNew feature or request

optimizer

triaged

Status: Open.

#1257 In pytorch/ao;

· msaroufim opened on Nov 9, 2024

[MX | Triton] Create MX matmul op using new <code>scaled_dot</code> op in Triton

enhancementNew feature or requestNew feature or request

good first issueGood for newcomersGood for newcomers

mx

Status: Open.

#1084 In pytorch/ao;

· drisspg opened on Oct 15, 2024

Add weight tensor-wise scaling for INT8 quantized and mixed-precision training

enhancementNew feature or requestNew feature or request

good first issueGood for newcomersGood for newcomers

triaged

Status: Open.

#1010 In pytorch/ao;

· gau-nernst opened on Oct 4, 2024

Add MyPy support

enhancementNew feature or requestNew feature or request

Status: Open.

#856 In pytorch/ao;

· drisspg opened on Sep 9, 2024

[RFC] Add LayerSkip to AO

enhancementNew feature or requestNew feature or request

Status: Open.

#633 In pytorch/ao;

· jcaip opened on Aug 8, 2024

torch.iinfo() support for sub byte dtypes

enhancementNew feature or requestNew feature or request

Status: Open.

#308 In pytorch/ao;

· vayuda opened on Jun 3, 2024

Generic packing algorithms from size N to M

enhancementNew feature or requestNew feature or request

Status: Open.

#284 In pytorch/ao;

· vayuda opened on May 26, 2024

FP6 dtype!

enhancementNew feature or requestNew feature or request

Status: Open.

#208 In pytorch/ao;

· NicolasMejiaPetit opened on Apr 28, 2024