Does default package support ONNX?

[NikilXYZ]

The docs are a bit unclear - what version should I download for using with Torch, ONNX, and CUDA 12.4?

Do I need to download multiple versions?

[NikilXYZ]

And relatedly, what if I want to use the latest version of torch? Do I have to downgrade to 2.2.2 or 2.1 instead of (as of today) 2.5.1?

[quic-bharathr]

Hi @NikilXYZ sorry about the confusion. Our latest documentation attempts to provide clarity on appropriate AIMET package to install: https://quic.github.io/aimet-pages/releases/latest/install/index.html

1. To install the latest version of AIMET that is compatible with the new Torch versions, see: https://quic.github.io/aimet-pages/releases/latest/install/quick-start.html
2. For other variants of AIMET (TensorFlow, ONNX and specific PyTorch versions / GPU / CPU environments), see: https://quic.github.io/aimet-pages/releases/latest/install/index.html#alternative-packages

You need to download one AIMET version for each framework that you want to use AIMET with. The supported environments for each package are documented on our releases page: https://github.com/quic/aimet/releases