NVIDIA / NVIDIA/Model-Optimizer

How to do sparsity and quantization-aware-training ?

Open
#122 3 comments 0 reactions 1 assignee View on GitHub

@kaix-nv is already working on this.

Since Nov 6, 2025.

feature feature request investigating torch.quantization
Dominant language
Python
Stars
3.8k
Forks
604
Avg merge
2d 8h
Merged PRs (30d)
142

Description

How to implement sparsification and quantization-aware-training , and keep the sparse mask unchanged during quantization.

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.