NVIDIA AMMO documentation

[fedem96]

Is there any official documentation of NVIDIA AMMO toolkit? If so, where is it?

In particular, I'd be interested in documentation about:

• implemented features
• supported quantization techniques for each model type
• changelog between versions

@Tracin @juney-nvidia

[juney-nvidia]

@RalphMao do you have any comments on this ask? :)

[dmitrymailk]

Same. How I can find source code of this library? I want to write custom quantization pipeline for encoder-decoder models like T5.

[yao-matrix]

same question here.

[puppetm4st3r]

x2

[ChristianPala]

Hi folks!
Are there updates on the docs?

[lix19937]

+1

[RalphMao]

Hi all, thank you for your interest. The AMMO toolkit has been renamed to "TensorRT model optimizer" and the documentation is available at https://nvidia.github.io/TensorRT-Model-Optimizer/ . Examples related with Model Optimizer is available at https://github.com/NVIDIA/TensorRT-Model-Optimizer?tab=readme-ov-file

[RalphMao]

Same. How I can find source code of this library? I want to write custom quantization pipeline for encoder-decoder models like T5.

The library is available on PyPi, with source open (instead of open source). You can access most of the files but some files doesn't have the approval for open source (yet)

[ashwin-js]

Hi @dmitrymailk , I am also exploring ways to run 4bit quantized encoder - decoder model in tensorrt-llm. Where you able to make any progress on that front ?

[github-actions]

This issue is stale because it has been open 30 days with no activity. Remove stale label or comment or this will be closed in 15 days."