nvidia-modelopt

View on PyPIReverse Dependencies (1)

0.17.0 nvidia_modelopt-0.17.0-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
nvidia_modelopt-0.17.0-cp39-cp39-win_amd64.whl
nvidia_modelopt-0.17.0-cp39-cp39-manylinux_2_28_aarch64.whl
nvidia_modelopt-0.17.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
nvidia_modelopt-0.17.0-cp38-cp38-win_amd64.whl
nvidia_modelopt-0.17.0-cp38-cp38-manylinux_2_28_aarch64.whl
nvidia_modelopt-0.17.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
nvidia_modelopt-0.17.0-cp312-cp312-win_amd64.whl
nvidia_modelopt-0.17.0-cp312-cp312-manylinux_2_28_aarch64.whl
nvidia_modelopt-0.17.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
nvidia_modelopt-0.17.0-cp311-cp311-win_amd64.whl
nvidia_modelopt-0.17.0-cp311-cp311-manylinux_2_28_aarch64.whl
nvidia_modelopt-0.17.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
nvidia_modelopt-0.17.0-cp310-cp310-win_amd64.whl
nvidia_modelopt-0.17.0-cp310-cp310-manylinux_2_28_aarch64.whl

Wheel Details

Project: nvidia-modelopt
Version: 0.17.0
Filename: nvidia_modelopt-0.17.0-cp311-cp311-win_amd64.whl
Download: [link]
Size: 1094628
MD5: 2c8ab19d35783a0534230a5d15b30d8c
SHA256: f62ffbb4cdfb86a6b965ec5d20ef0709520074fcc5f6be912f540322c527873b
Uploaded: 2024-09-11 15:39:02 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: nvidia-modelopt
Version: 0.17.0
Summary: Nvidia TensorRT Model Optimizer: a unified model optimization and deployment toolkit.
Author: Nvidia, Inc.
Project-Url: Homepage, https://github.com/NVIDIA/TensorRT-Model-Optimizer
License: NVIDIA Proprietary Software
Classifier: Programming Language :: Python :: 3
Classifier: Operating System :: OS Independent
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Intended Audience :: Science/Research
Requires-Python: <3.13,>=3.8
Requires-Dist: cloudpickle (>=1.6.0)
Requires-Dist: ninja
Requires-Dist: numpy (<2)
Requires-Dist: packaging
Requires-Dist: pydantic (>=2.0)
Requires-Dist: rich
Requires-Dist: scipy
Requires-Dist: tqdm
Requires-Dist: setuptools; python_version >= "3.12"
Requires-Dist: cppimport; extra == "all"
Requires-Dist: onnx (<1.16.2); extra == "all"
Requires-Dist: onnxconverter-common; extra == "all"
Requires-Dist: onnxmltools; extra == "all"
Requires-Dist: onnx-graphsurgeon; extra == "all"
Requires-Dist: onnxruntime-extensions (>=0.10.1); extra == "all"
Requires-Dist: accelerate; extra == "all"
Requires-Dist: diffusers; extra == "all"
Requires-Dist: huggingface-hub (>=0.24.0); extra == "all"
Requires-Dist: transformers; extra == "all"
Requires-Dist: pulp; extra == "all"
Requires-Dist: regex; extra == "all"
Requires-Dist: safetensors; extra == "all"
Requires-Dist: torch (>=2.0); extra == "all"
Requires-Dist: torchprofile (>=0.0.4); extra == "all"
Requires-Dist: torchvision; extra == "all"
Requires-Dist: onnxruntime-gpu (==1.18.0); (platform_machine != "aarch64" and platform_system != "Darwin") and extra == "all"
Requires-Dist: onnxruntime (==1.18.0); (platform_machine == "aarch64" or platform_system == "Darwin") and extra == "all"
Requires-Dist: coverage; extra == "dev-unit"
Requires-Dist: pytest; extra == "dev-unit"
Requires-Dist: pytest-asyncio; extra == "dev-unit"
Requires-Dist: pytest-cov; extra == "dev-unit"
Requires-Dist: pytest-timeout; extra == "dev-unit"
Requires-Dist: toml; extra == "dev-unit"
Requires-Dist: tox (==4.17.1); extra == "dev-unit"
Requires-Dist: tox-current-env; extra == "dev-unit"
Requires-Dist: onnxscript; extra == "dev-unit"
Requires-Dist: accelerate; extra == "hf"
Requires-Dist: diffusers; extra == "hf"
Requires-Dist: huggingface-hub (>=0.24.0); extra == "hf"
Requires-Dist: transformers; extra == "hf"
Requires-Dist: cppimport; extra == "onnx"
Requires-Dist: onnx (<1.16.2); extra == "onnx"
Requires-Dist: onnxconverter-common; extra == "onnx"
Requires-Dist: onnxmltools; extra == "onnx"
Requires-Dist: onnx-graphsurgeon; extra == "onnx"
Requires-Dist: onnxruntime-extensions (>=0.10.1); extra == "onnx"
Requires-Dist: onnxruntime-gpu (==1.18.0); (platform_machine != "aarch64" and platform_system != "Darwin") and extra == "onnx"
Requires-Dist: onnxruntime (==1.18.0); (platform_machine == "aarch64" or platform_system == "Darwin") and extra == "onnx"
Requires-Dist: pulp; extra == "torch"
Requires-Dist: regex; extra == "torch"
Requires-Dist: safetensors; extra == "torch"
Requires-Dist: torch (>=2.0); extra == "torch"
Requires-Dist: torchprofile (>=0.0.4); extra == "torch"
Requires-Dist: torchvision; extra == "torch"
Provides-Extra: all
Provides-Extra: deploy
Provides-Extra: dev-unit
Provides-Extra: hf
Provides-Extra: onnx
Provides-Extra: torch
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 102 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (74.1.2)
Root-Is-Purelib: false
Tag: cp311-cp311-win_amd64

RECORD

Path Digest Size
modelopt/__init__.py sha256=t4cVfYKBeLUteTaxDlZl2sIXnM4AaAfKFFRfUA0XbsA 694
modelopt/deploy/__init__.py sha256=kPLOuoLCaV0xwzl2Me4kXwJECUXDIa_YR2ozADCnYbQ 604
modelopt/deploy/llm/__init__.py sha256=RWZ04Y4QVWJry5cB_1CEbJl2hW0Mk1SKmJwoND62AIE 1059
modelopt/deploy/llm/generate.py sha256=82rlddo8PECyVVSj2dhrizcNSEPzipSvmnH3TdYyVo0 11521
modelopt/deploy/llm/nemo_utils.py sha256=yFA7zVsJdph2FV2eJmU3-qKhrlKq4ccozHDwb7E9Ew8 7332
modelopt/onnx/__init__.py sha256=2FUcmfR-kxw22Gd3a7cFiQrE6BjiUHZ7ZtXlYW0INE4 791
modelopt/onnx/op_types.py sha256=lxNs3RMK5yEb5BiPzfMPwkuNmQofSw6OXHQqQvakh94 7207
modelopt/onnx/utils.py sha256=oa4z4OGDI9H1OX1AlsFK9fMpoS7_s8L_wASXF3GuKJM 20312
modelopt/onnx/quantization/__init__.py sha256=mU2p7lDQXgRlNeKlbr7pG1_ouDY7M3p1TUFpWMFlba0 653
modelopt/onnx/quantization/__main__.py sha256=DDIc4Hc1esAcJwIA10OohGMqJxqfo1cMmnxog-d-Nfc 7091
modelopt/onnx/quantization/calib_utils.py sha256=AlH0wVVqgXfQY3m3mK4F_LME4Hm3Lz81K9t4eTJ-_-I 5552
modelopt/onnx/quantization/extensions.py sha256=109fj5mqFYa_OuWR-a3ZSBXV0thvdjLD4u9LaPTQ-h4 1109
modelopt/onnx/quantization/fp8.py sha256=zACQLjouNf-5jXjksGMu54HLoaVR7GtMmgcSY09ffYk 9913
modelopt/onnx/quantization/graph_utils.py sha256=WteW_f8NRcuaAkonlT3t6RCrFN44kT6TJi-OH_WQjqg 36790
modelopt/onnx/quantization/gs_patching.py sha256=vfuLeEz5ztIEqJ4G2DfBIWk7vquuKm2yKEy7Aa_IWdA 4124
modelopt/onnx/quantization/int4.py sha256=H4nocpkMvjA6JHTd7koilf0ckqygCMRgS2FDseFQDCI 31134
modelopt/onnx/quantization/int8.py sha256=f8h0ku4YOVetxR1QGy5V0f19YmDt7fVC05WQIMPi69A 9627
modelopt/onnx/quantization/operators.py sha256=GjExg_R2dyj9hsIxnGPiDt-0GJgQeR3qhgLsuLJFSaI 4345
modelopt/onnx/quantization/ort_patching.py sha256=JWkyLRgtJK2hllDfvwOMiHCyoDREkB1mp5tiVa2hSCg 25842
modelopt/onnx/quantization/ort_utils.py sha256=fPXjT2CFz4b4SZYFVePIgz_sg_DCE145AVsvbHhL_fE 4393
modelopt/onnx/quantization/partitioning.py sha256=-qMR6fAeiYJ4kcZpixPwLOJYEeovdB8Ypkd0axJGcAY 14054
modelopt/onnx/quantization/qdq_utils.py sha256=I1U0t7MMMEchvp4998bCKoQwOaXh4QkEsGRAb5xAIoQ 19196
modelopt/onnx/quantization/quant_utils.py sha256=0ADaN88ern-tyr7GeI0eN8FlXto3o59zAPPetJJ-NaQ 4839
modelopt/onnx/quantization/quantize.py sha256=yCmUZLRg7Gt24xW4j-T-AW9sy5tFe3h9zf228C_fq-4 15949
modelopt/onnx/quantization/src/modelopt_round_and_pack_ext.cpp sha256=irvtdVP_YvQWil-U66B52hQJTWptTlBN44g2MUsDQMU 4210
modelopt/torch/__init__.py sha256=8TP9ikcztqwi5VAdBeDOAuX0camq_2EZBfhz3LNxWKo 841
modelopt/torch/_deploy/__init__.py sha256=wLswmGXJx9LEBt7qspaRgfZ85WV3PhlOJgZErB5l1K8 876
modelopt/torch/_deploy/compilation.py sha256=D_n0nK37U_5wXy780gne1XLgVd5Q162lnYWNwDTiRMw 4847
modelopt/torch/_deploy/device_model.py sha256=Erk23_sh8R9y60J_YDchQyNS6KC0b5q3B3JFA00ovAQ 5522
modelopt/torch/_deploy/profiling.py sha256=aO-T_jtNHEk0M9whkFw-aWaq4_bRnIf0jeeTsGLrhcs 7588
modelopt/torch/_deploy/_runtime/__init__.py sha256=xVnt845HcVeoBK9RxvivmyEXkROiDrx6gIELsXDEJAE 957
modelopt/torch/_deploy/_runtime/common.py sha256=vzqRT2DHbTchiJddXdcrm3TTeYwJ3ZjwWDFeFYKXbIY 2327
modelopt/torch/_deploy/_runtime/ort_client.py sha256=wVLcgNPFE5xTCDY77gd4bvWYS72QeZIaiEiR4qIPTys 7266
modelopt/torch/_deploy/_runtime/registry.py sha256=4CTHLZZbQTRu13PFV2QIgqhRPW6gWZ2ftnUERz5v_ec 2671
modelopt/torch/_deploy/_runtime/runtime_client.py sha256=DQpJfwPHdXCB6EiSHbbepvkCqP0aP7yqyGRzCKG0q7k 5865
modelopt/torch/_deploy/_runtime/trt_client.py sha256=yge6HXXxvqVsWvY5NuJoOZDeBgf6cnON3y-pAXhMvBQ 8095
modelopt/torch/_deploy/_runtime/tensorrt/constants.py sha256=EIAT3YC-BUf402kP2su1K5r0FHyJ1XjyK7O_WB5cyCM 2807
modelopt/torch/_deploy/_runtime/tensorrt/engine_builder.py sha256=4ZerKdltMiswjoH8ZivpphgbdUlL9n8hCkgcNqn4t-M 11822
modelopt/torch/_deploy/_runtime/tensorrt/hw_param_config.py sha256=Om3X9ryKTsqRFWPN0vkdoSDq9DV7wO0-ErVeyFZIWQI 1763
modelopt/torch/_deploy/_runtime/tensorrt/layerwise_profiling.py sha256=s2-8XGsrlzTtkWVSfZlFNptlyEGQ3gLL0vH8tiJEEn0 6972
modelopt/torch/_deploy/_runtime/tensorrt/parse_trtexec_log.py sha256=k2DcHME2FYj61QYDuI0-hdEaAYWGm0jDgdShNW0jOp8 5691
modelopt/torch/_deploy/_runtime/tensorrt/tensorrt_utils.py sha256=lWQpy4eFQ7v4wQIT9aPQdIrM2aC2hg6zJasoFUxymcc 6524
modelopt/torch/_deploy/utils/__init__.py sha256=IsziBUwYpHo5BDrPdakSq7qYp-UJbiEPZ6i957NSf6A 633
modelopt/torch/_deploy/utils/onnx_optimizer.py sha256=i36vl83-Np_WLrpzh28whxSqWf2LFt246NvXtQolo_c 4730
modelopt/torch/_deploy/utils/torch_onnx.py sha256=0RgiWeYPKVUxIP0p8FV9kDZhWJTq2qUoPeLuPoDmRoM 23033
modelopt/torch/distill/__init__.py sha256=doESWwii6Tm4QU9k8wrdqEM8lVJ9CzW8YoWnTKVbNAs 798
modelopt/torch/distill/config.py sha256=Ql0uyMavnbVXg9leO8MDKrSCMJhniZDiAMP7aep55NA 5107
modelopt/torch/distill/distillation.py sha256=9VbHVysZDaE1nvIG017ahxcp9YGLgktGY4z_U74xDmw 3255
modelopt/torch/distill/distillation_model.py sha256=GJpd0P54UfzX6OJWPB6j8vPgOaCUzMQSGTWYxnA1WZA 11592
modelopt/torch/distill/loss_balancers.py sha256=aT5m9UGoHLMxxrW3c48jm7B3KrNb0wf-pAXKsnmR-9o 5582
modelopt/torch/distill/losses.py sha256=4w9H-tIt4FFVBBXBbd2P4XpBaklebIqc2_wP_BBI-2k 5131
modelopt/torch/distill/mode.py sha256=KI23t0KrmEvuP0T7jAQqhHwhB_hhbjdFkV5Xwt9JOWc 8181
modelopt/torch/distill/registry.py sha256=PkBzUk40V87zzxSbcpmiocVLS2q-XWNuq0WO76bPDKQ 879
modelopt/torch/export/__init__.py sha256=wKsAzMTU_u3JeoN-a6ghXhkNbuEGXikrBoOgQaQeyec 1104
modelopt/torch/export/distribute.py sha256=0Z3teFBuXCvDIS--4ZWbhf0x4FvdysAWqjXCkJ4R2W0 11680
modelopt/torch/export/hf_config_map.py sha256=JemRiTPjV1tKUzi9jzS2RCwisiy82mssvsiWtZBnmmM 3838
modelopt/torch/export/layer_utils.py sha256=j6kas8f-pKrUGEMjm2Qazr166zMh9JTPcUZIAhrRSAU 63354
modelopt/torch/export/model_config.py sha256=B3h6QvJv4oWo8YeiWu2O06HgCludt7_FjKK83DyUT2c 16870
modelopt/torch/export/model_config_export.py sha256=FUtpYlKaRrjsuOscB85EVBux8hlIgRrREjTKBCKNUE0 23689
modelopt/torch/export/model_config_utils.py sha256=fWhibx85IrppoEcpBmXsRys8BMmisyCceo4XE7jJSIA 20418
modelopt/torch/export/postprocess.py sha256=Yr5OTUyEdSOqe7UcOmiVMEF0ERdjYnW_tEuBLqntGNM 32156
modelopt/torch/export/scaling_factor_utils.py sha256=pcxqQE6l7wgdDJY1nqrDZ1EoHdb9vtr5lo6kn94R250 5549
modelopt/torch/export/tensorrt_llm_type.py sha256=Ak28TTwc_3FTQIn6EWgLcwMNshfuA229-qyt_WEJovY 1205
modelopt/torch/export/tensorrt_llm_utils.py sha256=9JH2NyGprXBlErQ6IPAgq-tPeQAISfDAyhD24zovSCs 24916
modelopt/torch/export/transformer_engine.py sha256=FXbHlEY5jkzLpWaQeeHqqBhdie7GZvdO0yFJ22FJDMw 4088
modelopt/torch/export/unified_export_hf.py sha256=0ZaymEliYgmmwzPZKfLi7r9ooJot1f8ZqyDxwZPXCjQ 9102
modelopt/torch/export/vllm.py sha256=faIJO9M-JgNrQFWVzhYSbrqEoTwHaJy2ZI9_4cof4LM 4270
modelopt/torch/nas/__init__.py sha256=gA5YUtnSUXTfywqPJvg3I1VOzGn0KiwGxpeIChWADdk 926
modelopt/torch/nas/_algorithms.py sha256=cwIuiaRHXahCnZVj9wbChr_iPONysT-yGP67_H-ADNM 18004
modelopt/torch/nas/_patch.py sha256=7_RM4tgXAwFW2uq1trWbRmUH0nOzxG4fJM7efGgfnwU 10384
modelopt/torch/nas/algorithms.py sha256=yGMJI7q748-NBP0S7tJ9SY4xdjZEq3_U547Eyn3j2zQ 19575
modelopt/torch/nas/autonas.cp311-win_amd64.pyd sha256=aRlR_Io0nJYMKUU0OifyS_wbJSXYmVuRNmg0pKOmdDc 244736
modelopt/torch/nas/config.py sha256=ms_R_Iw7f-jpPQHZXjp_F_VELtf2_oqyNhyc7KMOtuw 2899
modelopt/torch/nas/conversion.py sha256=lgYNDn2B93-fbxtmCYNLO0R6h8tMFYVQ4NzK9Y-veXY 5403
modelopt/torch/nas/mode.py sha256=l50YdOeLlHk-mLhHzsRkYyQTZLFT_nJXybfy-vsO-FU 4267
modelopt/torch/nas/registry.py sha256=y32sKBoKlc75_o0RXlwck2PbfcZ_U8iGDVpN2rniZ0Q 829
modelopt/torch/nas/search_space.py sha256=2hNiaXVxwTYCO7qyJO5Xb-40YalYV-qj0LOOdEuZI_c 10447
modelopt/torch/nas/traced_hp.py sha256=9zGHdCd3Pd9cuKQJ9Tv3JGMK5iCgs18mNI1VCSFHpNc 5662
modelopt/torch/nas/utils.py sha256=RvlTUbq9W0WAmM2XnnBI1NFzdXlLLxNDc_EH-ml4MpI 13572
modelopt/torch/nas/hparams/__init__.py sha256=sIgv0n4V0yC3tM3fzb4-MmnRLVzqlxcq05U-tzNu9UQ 689
modelopt/torch/nas/hparams/concat.py sha256=AnuM2shjesaNrR8ytv4iccyWdade7QOI4bQKJmdGMsY 12704
modelopt/torch/nas/hparams/container.py sha256=bGa5vqmgo2sw55aOBvQGlClaWvimeSfdQ-fV0T-VbOY 1656
modelopt/torch/nas/modules/__init__.py sha256=63nHZlLfbC7CrHuLCvK_goD2lKLz76w10PGz6AfQLkQ 705
modelopt/torch/nas/modules/container.py sha256=Kps540zH1Oslm3D4PbDBZKooQeKoERwxBHJj8gul7Xk 3456
modelopt/torch/nas/modules/conv.py sha256=aaL-O94AAu2PPQgNPe3wmH0CYnP8k2986SoRrrKgMsQ 10753
modelopt/torch/nas/modules/linear.py sha256=HrMJex5ouZ8I2kbKfSfgp21qPt8mGcn7HjODTWEu7i8 3150
modelopt/torch/nas/modules/norm.py sha256=5U7tj8esUCyruk2lGrWOvJwRmnK0wBph5Sb1VDYg-A8 7651
modelopt/torch/nas/modules/utils.py sha256=EaCJoqZWaFZ_jA82szecx0YdZw-HXDM5yaE_MiJu5Bg 2053
modelopt/torch/nas/plugins/__init__.py sha256=JobGEHVnv1X8KyFnUQ0XbB86mcGMmFZvA09oQg9yVvA 1278
modelopt/torch/nas/plugins/megatron.cp311-win_amd64.pyd sha256=oKzBTvrkbigpY1ubX5o6bIbvWexCbKfO_WSw22iXiiY 264192
modelopt/torch/nas/plugins/torch.py sha256=3Oyl0e9Wa7p2RWABEbxFAgcO50SBAgdQ9rJfbUC39sQ 2510
modelopt/torch/nas/plugins/transformer_engine.py sha256=w-qaHX384mODqsJtW23eZoCgEuactbjG5QXtnPFGUbo 1538
modelopt/torch/nas/plugins/transformers.py sha256=8c9-JY4D2POTCHXVosaOT4-PX_-gpFANN8rO9jiufdE 8380
modelopt/torch/opt/__init__.py sha256=DQKIOr5vN-rwRSOOCyqcvspQUttXoOk4AM7A3Uc1OSE 1419
modelopt/torch/opt/_hooks.py sha256=PVO8kSihdf46nc-8zU5oqFU3lwm1V68CkNOlgrEFKHQ 3137
modelopt/torch/opt/config.py sha256=CBeNoDb6qFw5mgFu4Nl-BLaPCLWsfVFK30sOjeTpz-M 15227
modelopt/torch/opt/conversion.py sha256=-1mLUwIt725nc3kOz8MK1h4okWv8LLSqd4dydR0rrUE 24062
modelopt/torch/opt/dynamic.py sha256=AwRQ2ghIJJn30Pt3W4BrRYtNFJnCg7Ttr0LzAqkxxKU 59249
modelopt/torch/opt/hparam.py sha256=gVbS-UAt2raCaYRiuHNzcHgGyHBb01W8UcvMlTiPJp0 10188
modelopt/torch/opt/mode.py sha256=9ys9j_HQ0uzUv7rWMQ4MvswMcRTyEDHap2iyNd_mpbI 12723
modelopt/torch/opt/searcher.py sha256=Y9u5ND7PZPQeIFvHEuWwZMGPn3VDAiyF4z1GO8sf5dY 14393
modelopt/torch/opt/utils.py sha256=xbfldf-xUGGKKFuZmQojcdMLk9J0VJz8wdZOxPH-EtU 2554
modelopt/torch/opt/plugins/__init__.py sha256=QNRSm7lfqXhwwZmFo_-M4fUyrkNU5ScNFJtZjJx-ZLc 862
modelopt/torch/opt/plugins/huggingface.py sha256=HIQM72hgb2knfCelfhhon35epEYn5pjsRCnWECCefEQ 5096
modelopt/torch/opt/plugins/mcore_dist_checkpointing.py sha256=of0IASOolo9MK1raa0e5Ib50iXJtZ_8hUvdkwLuOFlQ 15701
modelopt/torch/prune/__init__.py sha256=DP-6od-Hl6zNEfejKnS5Ntf5YJLVgCt6HKk2SfaxGQI 997
modelopt/torch/prune/config.py sha256=4OEfekSHvVk6pTdGpeQN64LivYm1bTVPiymqjlcm5Xs 2882
modelopt/torch/prune/fastnas.cp311-win_amd64.pyd sha256=2Hne03i1MN6W4pfAB8nIL0aLsQROsZJ4YiOTw9ax7QU 141824
modelopt/torch/prune/gradnas.cp311-win_amd64.pyd sha256=XZTlWgojCTVJTEeA5XnWNX8GcnFrNxgXmGlCW3OacuU 151040
modelopt/torch/prune/mcore_gpt_minitron.py sha256=IcLm1tA-5neQ_zHzS-VymK65DFDRd0xkSh44U1qExAU 5777
modelopt/torch/prune/mode.py sha256=CAqx4HXQEqpYCUKH1kIwHjXo74apaZyZ5I6N8qLjKEY 4831
modelopt/torch/prune/pruning.py sha256=S0uSShLMc2rYlofMfKNswrqWjgef9ou1SYyl3qpduXk 10737
modelopt/torch/prune/plugins/__init__.py sha256=oYw5XdUhQn4c2KgiNYrkaT7sCdwSQzKH5FJ5RUq44fA 1174
modelopt/torch/prune/plugins/megatron.py sha256=CNiu_gSudfdWqF2Sj5LMMCRoHh1PCZyN1pUEture5_o 1406
modelopt/torch/prune/plugins/transformer_engine.py sha256=EG9Gpx3tNmtxTfrCFk70XgMMz271fha__VyRGdLbkjk 1022
modelopt/torch/prune/plugins/transformers.py sha256=GVYFfgve-Fxcp9eUpAwbpmrGqQX9lR1EaCuOJ-emcrU 1194
modelopt/torch/quantization/__init__.py sha256=vNWH0Qf7KPHGhsoRZZiH20QkaW__rzDrqody_urFmXY 853
modelopt/torch/quantization/algorithms.cp311-win_amd64.pyd sha256=0K1S6kaw66KVUUTSCXUbaPs9aFB9NcQ9ZM7ZBhMKZE8 251392
modelopt/torch/quantization/config.py sha256=Nyv8dXOzTX847t0wY8HYMmGDu_TOHrMN3UWJgPxBgUI 31117
modelopt/torch/quantization/conversion.py sha256=A_5s8tGaiD0-41iXw419t9Ym7mDMUOb_5ZTBuedd3-4 13701
modelopt/torch/quantization/extensions.py sha256=vPZExmNXy4wIl6LLhyH3gTgpHfn3EbUCyDjOa1hZ5Ak 2444
modelopt/torch/quantization/mode.py sha256=OQcfuY9j8lJRmuMWKGca00HLpOz9ufgyTGoZaZf22j8 4062
modelopt/torch/quantization/model_calib.cp311-win_amd64.pyd sha256=JSlpMoRsJmxT4wfemMyPc7bf4YSWrVrywpeaGW_7MOg 373248
modelopt/torch/quantization/model_quant.py sha256=37iunio1FOPvx7PtCEkmsGh0y0HtsAd_t-IVnRz8orY 14384
modelopt/torch/quantization/optim.py sha256=SBYwxlDU8YQH8QyEq_GRG_8_NsqdOMoSKcAw3e00sMw 1393
modelopt/torch/quantization/quant_modules.py sha256=CI4kw5-cXgQBWotSOM36q3ZxJmlDzAWkT05k4tR5HKI 1725
modelopt/torch/quantization/tensor_quant.py sha256=ck54SrLz1XpmHLbFhJH7H_OyRHEhPZ8UvjWSKEmUHV4 23925
modelopt/torch/quantization/utils.py sha256=MM7--jEERrMAZoRBQ8omnGOxLTDiBtSZhE2YCYm5Lg8 6358
modelopt/torch/quantization/calib/__init__.py sha256=QC2CDk13q2ZbfENLeWk4s3ckC8FJBd7pTolMc44Sf4A 816
modelopt/torch/quantization/calib/calibrator.py sha256=Bt7iipG8jKTmUAvfUn_sl3GbTEzm6zdLYNszJgKiCuw 1886
modelopt/torch/quantization/calib/histogram.py sha256=GPWJkjQ_xJD0JpUphFmrZ1EmgPianx54Hjq-8AWVMKo 17497
modelopt/torch/quantization/calib/max.py sha256=uFrG65TXCH_9M2s9KlSBjQTnG2kuDwAYVhZtWbnD5h4 3490
modelopt/torch/quantization/nn/__init__.py sha256=GLwODQD5OwT8MppeDUvgjgotkEFyLWZTAy21_EJ0TQs 979
modelopt/torch/quantization/nn/functional.py sha256=1J5SB4T3DpC-KBn9AR473kSwa0zCTKGyT9xxb3eKrlU 2822
modelopt/torch/quantization/nn/modules/__init__.py sha256=hiV4Q5QP0y1rl1zOgawSoknb6I-drlZ-Xr6EUmiwyeE 601
modelopt/torch/quantization/nn/modules/clip.py sha256=pyjEsnYkb87XQErWPsRDv8qBRFH09PC3_5fQjygutcw 1839
modelopt/torch/quantization/nn/modules/quant_activations.py sha256=MxRlVUZUT117uqpLEjAHUNUpdsZu-MmoOo5iZnIncNA 766
modelopt/torch/quantization/nn/modules/quant_batchnorm.py sha256=2BBZUQypVsvhC8FQO0CONHJf3jeDECv7HtLppSmL7Eo 942
modelopt/torch/quantization/nn/modules/quant_conv.py sha256=QMNBQRk1tPD6YbgiDMfAyWI7bWXfg-jAiGW9fij3bjo 3986
modelopt/torch/quantization/nn/modules/quant_instancenorm.py sha256=R2Qr4OFDdCaZLzRX8jlcJGL_kmVujLW01hlJDwopfHg 1556
modelopt/torch/quantization/nn/modules/quant_linear.py sha256=z6OzLpBStdH4GEdiGdzTf1-HInHSyLBwHlfNCc4nx4U 1676
modelopt/torch/quantization/nn/modules/quant_module.py sha256=lCT26ah_ahPdAtJK-jbK2U-DwcozdnU10jksh7KidI4 7130
modelopt/torch/quantization/nn/modules/quant_pooling.py sha256=mQfjb0Hao7wKxh4SgrE6_ll4aeVuqE8MA2nM_5IAE8g 3556
modelopt/torch/quantization/nn/modules/quant_rnn.py sha256=p9HHRc2LvMylzphXW-00kvt1mv6L7KaeAttcBCc3bFk 19262
modelopt/torch/quantization/nn/modules/tensor_quantizer.py sha256=yrYnqFtCdsu1C-MZ8wyImWhBMi347Z0dgZzKCFiUcQE 39000
modelopt/torch/quantization/plugins/__init__.py sha256=qGfcWjQGBkV5k6VD2MAXIPMtvLkazKakiGpGpxmwwNM 2130
modelopt/torch/quantization/plugins/apex.py sha256=gbmdZNtsGx-W1kyTTmYXUzEK8K-XHGHthhDONrK3l-g 1935
modelopt/torch/quantization/plugins/attention.py sha256=Ppn0pfubYMovcojVBRv33FbuTZjHQ3sruiFG8diNPWc 8337
modelopt/torch/quantization/plugins/custom.py sha256=DWE823oxW9tw0H8xTZwrNqaKEbG9AVuv87mC2dpoX2g 4398
modelopt/torch/quantization/plugins/diffusers.py sha256=E2mmptj822QIV9IULVbBX7VWjCE82KlrOHZYogiuzK8 3314
modelopt/torch/quantization/plugins/fairscale.py sha256=ihwtkIGdxzT5HqYBcXlOYy4XLs5Xk7AQ-wSr7xWS0dQ 1609
modelopt/torch/quantization/plugins/huggingface.py sha256=dwsquZ79qB_S91udbjFX0KZFBh10egCgdZq2ZxOC3dE 10649
modelopt/torch/quantization/plugins/megatron.py sha256=joUyCVH7W-X1BiDFHMKOeWqvzS_fhTgg6GjA7QsaDww 5275
modelopt/torch/quantization/plugins/nemo.py sha256=EN21xNTtlY0EVQeeZomMrpSI7V_V-twt0Hg8odB6YzQ 2472
modelopt/torch/quantization/qtensor/__init__.py sha256=1qfcgbT8Ef__xoMQVxHO4xPWbNWMUSFzqK-R3mAcSJ4 686
modelopt/torch/quantization/qtensor/base_qtensor.py sha256=Jzvj3fGCGwn0P3vCpuNkw2x1tra4ewtjv1wmZE90Mrc 3080
modelopt/torch/quantization/qtensor/int4_tensor.py sha256=dp1B2G01Qxi7Dcsatc-ZTiC26Y1i1Lxq84IqtMn_hJw 4852
modelopt/torch/quantization/qtensor/nf4_tensor.py sha256=VWJs99_5-gVPGW7tuGcU0Jgnp0Z5QZXMDMeF4dB2eUA 7486
modelopt/torch/quantization/src/tensor_quant.cpp sha256=vEWBk3hdAG5NYJhRqls5eI92tR8_41wPw_PlcYkpjHc 3843
modelopt/torch/quantization/src/tensor_quant.h sha256=AP0iYU-Sfe9gL_0HEULrKQNzYFbV8Ze_sy_rHxq0dRM 2147
modelopt/torch/quantization/src/tensor_quant_fp8.cpp sha256=GlPYcFMJznyRrh5ikMn4Q5LqbsuJiis6QCeJpauO0I8 1757
modelopt/torch/quantization/src/tensor_quant_gpu.cu sha256=4h0DTsBUY18PZEyPYT1bxQ176DHSapU0YFR3e-DfgRA 15151
modelopt/torch/quantization/src/tensor_quant_gpu_fp8.cu sha256=_hBiv4CWyImc6OK3iNg0y8jGNzcP-bQ6DGjGulcCHbU 4590
modelopt/torch/sparsity/__init__.py sha256=HZ_QgP_YXBYrek8UTJ_25p5FLEV8QsfADksqI-6V1Ic 671
modelopt/torch/sparsity/config.py sha256=lndNOuurv-YHJTu-15W8YgvFwqxSmdZahaTAZ2OvWBE 1641
modelopt/torch/sparsity/magnitude.py sha256=GQC2-XHvKbLAjoC8z_6pyLQMBBIrN0sbDhGZCnWOg4M 5352
modelopt/torch/sparsity/mode.py sha256=9RJivfnNG-yF9b6lsn91r-0Zc_u37K8jRWmsq_oXmx4 6847
modelopt/torch/sparsity/module.py sha256=LVB7t5nNN0Fdh-2HcnzhgH8m6AaPx9s8B5NgId_jN4g 3588
modelopt/torch/sparsity/searcher.py sha256=pxu51_7PSoppge1DoaAsElCu5G7y_E3-RRg3A3BkCJk 3119
modelopt/torch/sparsity/sparsegpt.py sha256=DtGnz8ITse9L-2vZ8Qk7tubM3ZK_I7pjrVO79ymjGTY 10880
modelopt/torch/sparsity/sparsification.py sha256=MjGCfHmk3s4WDB5t62bCI6Bw5rVY2rCxuACHeCNRd88 5744
modelopt/torch/sparsity/plugins/__init__.py sha256=YSIFMOUX0q6gzG82Rf9xN6TOP7mTHwrbthWT86FSwnE 990
modelopt/torch/sparsity/plugins/megatron.py sha256=Pd77yaBAWzwpHRn2LwhFg1zYlH_gT4vP1vSx-4qCVsY 2762
modelopt/torch/speculative/__init__.py sha256=Tsq48EILmRaok0rk_sHQuKAfIqgTvNE7L-5x_PwV0BA 699
modelopt/torch/speculative/config.py sha256=iwA6kiwvHQ8j-9ngWGJ2GIBosDY1UTZF7EYPOVaMM4I 1052
modelopt/torch/speculative/mode.py sha256=RFRZURRwJrFbpCaW8MGO1agjjOdS3y4bvQH-Gs55-q4 1889
modelopt/torch/speculative/speculative_decoding.py sha256=YA7TZ-jTNFuTrGI4EZL7suPKcjj7BtG1yGG2OhYx8Ag 2095
modelopt/torch/speculative/medusa/__init__.py sha256=sMaAQa-eKf1FIZe4m7iJIRwAtHqHqb4vR8kUXBDuqQc 650
modelopt/torch/speculative/medusa/conversion.py sha256=-l0TLDru0zRHwxab_OMenxNmk8w5FCUzV0bQO7by_vQ 2150
modelopt/torch/speculative/medusa/medusa_model.py sha256=5TwJXizSW3aXP-woxseSoQXOnjyP6GcOjKjAocOBNJM 2315
modelopt/torch/speculative/plugins/__init__.py sha256=v7n2muLgmBhJA44QBw_pxHT_nJ1YdR5sT7A_1n8aHUU 1052
modelopt/torch/speculative/plugins/transformers.py sha256=XuaNIMvM-F_lSTBA-hAo9sFjv0af5jciicO9yjS84as 14080
modelopt/torch/trace/__init__.py sha256=DXJ7Ii5LMgHa9rLm_59JbUXiHt8Ax818w_iGuIkDJH4 832
modelopt/torch/trace/analyzer.py sha256=I4dScIH0gujK7pHLhrDAHvmrXTkdMz9IGHkeSpD8A7Y 64318
modelopt/torch/trace/symbols.py sha256=FqeoyR2C1pXWQo68tjtLnri5vXAa7KM7Pgtlc9k5M88 21991
modelopt/torch/trace/tracer.py sha256=1htFzLBfUl8Ehg3LonIwlkOzRjlNBCjkyPSDIm1kpqA 13440
modelopt/torch/trace/modules/__init__.py sha256=2d5VU3LmPTDjtGCiPaN8NVOlNIfK8UddtgfcFFOGPlY 685
modelopt/torch/trace/modules/concat.py sha256=_5iEXSiPtg0aFchFhzK2BWozLlUKmc8pPux1i3ZODkc 17774
modelopt/torch/trace/modules/nn.py sha256=8ZrtCbe51vQltsZyJs3BFcP97CMcg2qrhVK7IIGaXKs 5715
modelopt/torch/trace/plugins/__init__.py sha256=GfiG5IUmjIhYBb-AU3mThgbPiCy9m1H6025zj4wPemw 1180
modelopt/torch/trace/plugins/megatron.py sha256=eqotM7_6mdhYvRGzSkWqhFz4IiuOJ0fk8vPAGN6yUn8 3496
modelopt/torch/trace/plugins/transformer_engine.py sha256=PMsWZ5u8M_ONjaXo4GCAWqNudkWKAaAaN6RbSfLNGuk 989
modelopt/torch/trace/plugins/transformers.py sha256=rz2d_BDu5tSe1PJas9Fud1PD9cpoME1D67SmBFE4jOI 2172
modelopt/torch/utils/__init__.py sha256=_3Fi4SbTPAi3wRkkfZ8WCo7Jj-SkA5XfUnK9hCcEMxg 804
modelopt/torch/utils/_pytree.py sha256=AJC3qOtD4ljmGcUjIboGxmI9zmIvyC7sOIMlDsLRL2o 5582
modelopt/torch/utils/cpp_extension.py sha256=aDxgP_sml9rLkoFRsirSNNbvNJqPDU_eTNe6YxcPwVo 2419
modelopt/torch/utils/dataset_utils.py sha256=2rqlU8BnrSIlX936v9cjW6LXnyIHp3x26J_mVbDdWQ0 10542
modelopt/torch/utils/distributed.py sha256=Wy8B935p2GmFrhm-yhhAiDLYiSSuUD8h2IR--RZJ-_M 9330
modelopt/torch/utils/graph.py sha256=udBDNDEpv9TPPEsYZemll8-5sksCJthE99XPPAcWWUQ 3894
modelopt/torch/utils/list.py sha256=Gof1ysHiU5ZuYQdefa-XZBsfR1233sPj-2k58jutAuk 1951
modelopt/torch/utils/logging.py sha256=HaP6k7_CRFDpTU999CUeRN9udoCScsX_uIL1Wv_Kwt4 2999
modelopt/torch/utils/network.py sha256=KCzi03mUe8zjPoq98gKBVEH7KkGps1lmUyiXvN2TQy4 24568
modelopt/torch/utils/perf.py sha256=RXyy66dfDNMYcjGHP5RdJM_nVX_xO57Nzx5k0aXvrcs 2823
modelopt/torch/utils/random.py sha256=lIg_8WdFH8plq57tXMK0DYxNoBzR8W2nOR-Gqhu6PWA 5706
modelopt/torch/utils/tensor.py sha256=Zxcrtp5Y2wrjHN0tukVz1GDfggRZxI2QNldWrbWcCKU 1999
nvidia_modelopt-0.17.0.dist-info/LICENSE sha256=xBMtI6murYnjWr29V05XL75sviTU9Sq1XfxrjhWJ6TI 540
nvidia_modelopt-0.17.0.dist-info/METADATA sha256=Xbfi0scw_a83VY5s7ZLHdRrfHBCRsliiONafhlObxx0 3583
nvidia_modelopt-0.17.0.dist-info/WHEEL sha256=WRbBgi9HBHR6yzaPlBW1thcwhnq7JwQcoZZASdBlCwA 101
nvidia_modelopt-0.17.0.dist-info/top_level.txt sha256=IMo1VR4nTJCILUd5BAb4Pd5DmXMRn_4W9wobCeTfn5Y 9
nvidia_modelopt-0.17.0.dist-info/RECORD

top_level.txt

modelopt