| Name | Modified | Size | Downloads / Week |
|---|---|---|---|
| Parent folder | |||
| aimet_onnx-2.40.0-cp310-abi3-manylinux_2_34_x86_64.whl | 2026-09-21 | 85.0 MB | |
| aimet_onnx-2.40.0-cp310-abi3-win_amd64.whl | 2026-09-21 | 765.2 kB | |
| aimet_onnx-2.40.0-cp310-abi3-win_arm64.whl | 2026-09-21 | 731.1 kB | |
| aimet_torch-2.40.0-py310-none-any.whl | 2026-09-21 | 816.0 kB | |
| README.md | 2026-09-21 | 1.6 kB | |
| Version 2.40.0 source code.tar.gz | 2026-09-21 | 9.8 MB | |
| Version 2.40.0 source code.zip | 2026-09-21 | 10.6 MB | |
| Totals: 7 Items | 107.8 MB | 0 | |
2.40.0
-
New Features
- ONNX
- Add
encodings_to_onnx_qdqto convert an ONNX model plus encodings into a QDQ model (fcfc439) - Add
session_options_kwargsargument toQuantizationSimModelto set onnxruntime session options (c173926) - Accept an explicit
LlmTopologyinapply_spinquantandapply_adascale(00c6089, c548be0) - Enable FP8 (
float8e4m3fn,float8e5m2) quantization simulation on both CPU and CUDA, completing FP8 support alongside ONNX QDQ export (644c7bd)
- Add
- Torch
- Implement
set_weight_quantizer_to_nvfp4_int8API (05cc606)
- Implement
- ONNX
-
Bug fixes and Improvements
-
ONNX
- Fix calibration state restoration in
QuantizationSimModel(df58875) - Analyze LLM topology and run SpinQuant on
onnx_irinstead ofConnectedGraph(23633b1, 7b9aef6, 092a9dc) - Add all kv-cache inputs/outputs to
LlmTopology(68e24d2) - Drop ignored
connected_graphparameter fromget_decoder_block_boundaries(2d71c85) - Auto-upcast fp16 AdaScale blocks to bf16 (1f03b7c)
- Default to using all float outputs in
make_psnr_eval_fn(5c41452)
- Fix calibration state restoration in
-
Torch
- Fix device mismatch upon
.to()for qmodules with pre-quantized parameters (4111e41) - Fix fp4-int8 dynamo export bug (68588d4)
- Fix device mismatch upon
-
Common
- Exclude depthwise Conv/ConvTranspose from supergroups and fix false negatives in depthwise conv detection (7729c97, 2a6b723)
- Publish a single wheel per platform instead of
+cpu/+cuXXXvariants to the GitHub release (4fc12d8)
-