nvidia-modelopt

View on PyPIReverse Dependencies (3)

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

Wheel Details

Project: nvidia-modelopt
Version: 0.21.0
Filename: nvidia_modelopt-0.21.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Download: [link]
Size: 2571150
MD5: a31349ce95499ec61d6def271e50a819
SHA256: da91de0f8a11187f78d86c555adf5142247335992b3bec5e68b0960f00ef6f93
Uploaded: 2024-12-04 05:36:41 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: nvidia-modelopt
Version: 0.21.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: 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: cppimport; extra == "all"
Requires-Dist: onnx; extra == "all"
Requires-Dist: onnxconverter-common; extra == "all"
Requires-Dist: onnxmltools; extra == "all"
Requires-Dist: onnx-graphsurgeon; extra == "all"
Requires-Dist: accelerate (>=0.27.2); extra == "all"
Requires-Dist: diffusers (>=0.27.2); extra == "all"
Requires-Dist: huggingface-hub (>=0.24.0); extra == "all"
Requires-Dist: transformers (>=4.40.2); extra == "all"
Requires-Dist: peft (>=0.12.0); 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-test"
Requires-Dist: onnxscript; extra == "dev-test"
Requires-Dist: pytest; extra == "dev-test"
Requires-Dist: pytest-asyncio; extra == "dev-test"
Requires-Dist: pytest-cov; extra == "dev-test"
Requires-Dist: pytest-timeout; extra == "dev-test"
Requires-Dist: timm; extra == "dev-test"
Requires-Dist: toml; extra == "dev-test"
Requires-Dist: tox; extra == "dev-test"
Requires-Dist: tox-current-env (>=0.0.12); extra == "dev-test"
Requires-Dist: accelerate (>=0.27.2); extra == "hf"
Requires-Dist: diffusers (>=0.27.2); extra == "hf"
Requires-Dist: huggingface-hub (>=0.24.0); extra == "hf"
Requires-Dist: transformers (>=4.40.2); extra == "hf"
Requires-Dist: peft (>=0.12.0); extra == "hf"
Requires-Dist: cppimport; extra == "onnx"
Requires-Dist: onnx; extra == "onnx"
Requires-Dist: onnxconverter-common; extra == "onnx"
Requires-Dist: onnxmltools; extra == "onnx"
Requires-Dist: onnx-graphsurgeon; 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-test
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 (75.3.0)
Root-Is-Purelib: false
Tag: cp38-cp38-manylinux_2_17_x86_64
Tag: cp38-cp38-manylinux2014_x86_64

RECORD

Path Digest Size
modelopt/__init__.py sha256=vyCwL13A_SBpbIPDvXX_3np87kbZ2jx-EPLouSeB4rQ 892
modelopt/torch/__init__.py sha256=7_62HJTSPE5VD_kuTU5s5_ANN1ipLa4Vsk5YmOeUmv4 1264
modelopt/torch/utils/_pytree.py sha256=RfxyikkQq0CImyMIUXHO5vOC5rO5yoPrdQ8PSYCAP60 5454
modelopt/torch/utils/logging.py sha256=P9MialeqvQsZ8LrV_UjlRH3QzBHupUStVGYBGqKrjFk 3857
modelopt/torch/utils/__init__.py sha256=glGlGmgar95V7UMyVJzuXjMshQODHspsMIvF_M6GIp4 813
modelopt/torch/utils/tensor.py sha256=2svJtd1JL0DSFk0RRtdZuNUsATGvxzoJ9_kR6o1qkOE 1950
modelopt/torch/utils/import_utils.py sha256=B0mNeADS1Wd2ubX66luw2-zFtBsZgtjgWPJ8f22u5AM 1008
modelopt/torch/utils/dataset_utils.py sha256=fVjUT2JLQp_NMTw5QdaiyXXDUJPyfZnLD3mveqqygjs 11622
modelopt/torch/utils/cpp_extension.py sha256=2HbZovfk_Hc-zqg_v6wKkRzbaiVa_QPedxHsVSb4O-4 3375
modelopt/torch/utils/distributed.py sha256=twRXhd-qPHyX7ugdspZF7o0H8ffuCjnnsu1qH9rF6Zw 9794
modelopt/torch/utils/graph.py sha256=DvcnS3A2QxziJdlF-EqPwHIK4gUYn6GPsvQ1Qzb-nW8 3776
modelopt/torch/utils/random.py sha256=kk_wF3AF40To7eDifZWdtT7ltgd9VPlfN06rFBX7bCc 5553
modelopt/torch/utils/network.py sha256=123qiPhKK6tXzQ6t-ZjxHKdqF-rQn_-lZndoBNmhcTA 23966
modelopt/torch/utils/perf.py sha256=iRnyaaig1tnmgeR4rg2f6Mof5x8kcb9naQIc8L4vhlU 2739
modelopt/torch/utils/list.py sha256=CB7X-7eHy8Xskqq29Wmnuhu_aw6UmUNgiVbt9302izg 1899
modelopt/torch/quantization/__init__.py sha256=6nRpTHmv1I5D2AfInMhjmGgg4_QNlLoNxNAxg9HSQnI 807
modelopt/torch/quantization/quant_modules.py sha256=7i9aIsDASEZ6PLZKEEe5iqilgtVsr5C5F-w8X2JOLeY 1680
modelopt/torch/quantization/conversion.py sha256=MpMHPQvTXiYvMf0_uF9Lc9hSjk16g8aFX3LEbo12hkk 14242
modelopt/torch/quantization/tensor_quant.py sha256=ixg3w7eEKDbFF66ATmqCqQpWfP4FcEf3wKrBDYLpIc8 20440
modelopt/torch/quantization/export_onnx.py sha256=OR1qN1YFDnZDoqa9qVURbQsiPuzS58akgYRlERvsKnA 14451
modelopt/torch/quantization/extensions.py sha256=RAhLKEsuipvRCKA2Eu9UpU3L07qUVPrff5T0NX4uQH8 2386
modelopt/torch/quantization/mode.py sha256=PZj2NpCFsQ2Xr_MFbmiNWuieCuTbm3hIGZHtBYAJykY 3945
modelopt/torch/quantization/utils.py sha256=n2Mwgd0BUlTP3_ptVtGSvGLcEexPaJifhSQuI5oAzX8 6179
modelopt/torch/quantization/model_quant.py sha256=BI0puOOy6YkeryQrCH1_gwbI23S_sYWfy4YTJXstfqM 18470
modelopt/torch/quantization/model_calib.py sha256=0aiMp0rxlDhGn9sReZ32zMOynTfcajAWxdzXOg4DJn4 38841
modelopt/torch/quantization/config.py sha256=bW0qAD-VJcBLOypKDHcv-i2Yf_A_1N4Mmi6Z1cb3YqM 30756
modelopt/torch/quantization/optim.py sha256=5wZLVH9-ZpF7c9j2TLamzRkMsarfpXdgrDuRuPegP94 1360
modelopt/torch/quantization/calib/__init__.py sha256=F9eLUv37AdqH-u51y9m2a3n6Y4mLc8w34PqtlbOaFTE 797
modelopt/torch/quantization/calib/calibrator.py sha256=-lbBgYUPmQijn5beLtnUe_GZrv9GqumIixLjgiAc4qM 1828
modelopt/torch/quantization/calib/max.py sha256=qbo4zRVLS5WuOzrxZ31gN1FlUdOKtJf-dq-RtoMoAPE 3392
modelopt/torch/quantization/calib/histogram.py sha256=oQ4-swLHW1aewQk9cHy7npYaeWxrsPPa62aNGVoy8pE 17126
modelopt/torch/quantization/nn/__init__.py sha256=xxzSyrOKzqG1XIRKyCQFLAoU-CbwuFaq-qcsIcNGo4Y 957
modelopt/torch/quantization/nn/functional.py sha256=HC4l7XuhIpTJ3z3b5Gz-6YDvJ_JrPjCWAKp6P-8_JGQ 2757
modelopt/torch/quantization/nn/modules/quant_batchnorm.py sha256=komDih78HcqGGJ3TxhZzIqq2M5Qo8KqQxNwq6hjdPO4 923
modelopt/torch/quantization/nn/modules/clip.py sha256=iOmK5BPY5amugLRm3ICIwMWNIkHQLqGzhA9rHSe5R9k 1790
modelopt/torch/quantization/nn/modules/__init__.py sha256=DPuO-IBBsGIlTJ46v-EX4giAC-WMemru74lsPwWykpQ 590
modelopt/torch/quantization/nn/modules/quant_conv.py sha256=Uy6ligJm1rRzFhwV8gdgdX6j08-zkzjVFOpg39L_k3Y 3869
modelopt/torch/quantization/nn/modules/quant_instancenorm.py sha256=79cNyCjiNoPfkWx9IOenfvawKOThc02ByJVVtN19sG4 1516
modelopt/torch/quantization/nn/modules/tensor_quantizer.py sha256=oq_1-U60UxDxqhKa8iT6Rv2-k8J51Y2bu0i_GYAS3Ww 39242
modelopt/torch/quantization/nn/modules/quant_activations.py sha256=xRKx-v7h537lwWj_1uJpyzT7SEi-pWCLuOHQs5Pn6Kk 749
modelopt/torch/quantization/nn/modules/quant_rnn.py sha256=_zpu_2JifE2ArHw-NHC6pnbzyGeoYmQZWZ10oG1GXlQ 18734
modelopt/torch/quantization/nn/modules/quant_module.py sha256=bsTbdEQhOa2o70yOpFHrunBs4CGHRfg7MuMlOVCQ5Uk 7660
modelopt/torch/quantization/nn/modules/quant_pooling.py sha256=77GWUXmg_mIi0ceKs5I2yo53LKH9Lj2zEwSNlinPa8w 3444
modelopt/torch/quantization/nn/modules/quant_linear.py sha256=bTwT0MyfvJDtpjyAu0ZREqnef2b-h-ZkE1UcrKFb8zQ 1628
modelopt/torch/quantization/qtensor/int4_tensor.py sha256=KqXbDASlQ8dNBQwOXLDMm1pbC64CARgN_9ceEWIPDk0 4801
modelopt/torch/quantization/qtensor/__init__.py sha256=scigOCyO4PFXfaxLcIOMfcv2KmgeUtXHfFKJAzfxzc4 701
modelopt/torch/quantization/qtensor/nf4_tensor.py sha256=gDG27FdWx72O9gPL3Twt3r4dLPeVTCd9Muj_ybEOURQ 7369
modelopt/torch/quantization/qtensor/nvfp4_tensor.py sha256=7Ayt_AOLChfvIa9DNTTHqxXcUfG5p1BlvYyrXNSYES0 2390
modelopt/torch/quantization/qtensor/base_qtensor.py sha256=9plkwMKTO5hAWBgQNzPq6BkkIZmXGmZB6JCWrE_67D4 6066
modelopt/torch/quantization/plugins/megatron.py sha256=ub0rCN4Y9-ELuMV6jkoV-4fgZf-_urOWIhitDcclQcM 12903
modelopt/torch/quantization/plugins/__init__.py sha256=SBYVAbM1GZjwb7Wv6xclvoddN-Zg15m3-4ONR8Wds8s 1683
modelopt/torch/quantization/plugins/peft.py sha256=e6t9Q3tlw17oukOMyTRStQEEhk6z2AWGJbdsRUabrwY 2040
modelopt/torch/quantization/plugins/apex.py sha256=-9aSbaw-g3cjeJnVPKljdrEwBQNz372r57QpW_NKTfA 1882
modelopt/torch/quantization/plugins/diffusers.py sha256=3GJ-mi0wSLgQqFDGsUC-5k93x7EYF6sR2jIGje9LPbU 7656
modelopt/torch/quantization/plugins/huggingface.py sha256=zaBci7wt174Ak-og7dElRF_PaqlYowsn752l1QlZOLU 10434
modelopt/torch/quantization/plugins/custom.py sha256=ercntGWnphfnw6SLZMqhY44Ama1TpOdk3spBx1I6YX4 4293
modelopt/torch/quantization/plugins/nemo.py sha256=E3WYxRC5Ii_x2jH9x3szR1QbRQxKxPTHwQGL_BEXwe0 2412
modelopt/torch/quantization/plugins/attention.py sha256=5irqkpKnZIJ8pcxRYdQOs-h-cxboiOLes61Pyv8jSPE 9796
modelopt/torch/quantization/plugins/fairscale.py sha256=Ha-K_-n11Rdf4Wz_ObEy74GhbYXpHsW4bvy7IAlZ42U 1570
modelopt/torch/quantization/src/tensor_quant.h sha256=L31GWBqWdhObZWFzZRyKxCe4L4QCsKayF-Cg3MUFmh4 2095
modelopt/torch/quantization/src/tensor_quant.cpp sha256=f__DidjjlkeaBQYl8geS1FDfe7FZ0w6fxywq8-DEifY 3771
modelopt/torch/quantization/src/tensor_quant_gpu_fp8.cu sha256=8iITnq8qhF4uGyk5zAMr3zzfiG5Oa0H5mjcoQOJryIo 4473
modelopt/torch/quantization/src/tensor_quant_gpu.cu sha256=BRH3Hi8SfRPnBmChN35txU4F6qfLDnU6JENGpc4NZPA 14795
modelopt/torch/quantization/src/tensor_quant_fp8.cpp sha256=R1HkSx5r61voY3xyH-S-c76QvCNG6RR379X0hHV8vh8 1714
modelopt/torch/sparsity/sparsegpt.py sha256=lWWLSnEeZ21EJAftdD-4zu6HgKumXAv40qRjBa2DZfc 10608
modelopt/torch/sparsity/__init__.py sha256=W-aGBn5SiPIBjB4KaMa32JbT5psrQzSKM7ko3cAZJrc 657
modelopt/torch/sparsity/sparsification.py sha256=zR1wz2ZqsWThGhHjbTYcZXA4t4urofUA2Yw0Xcm7iAs 5628
modelopt/torch/sparsity/magnitude.py sha256=o_F5TQ0WDj_5MfxiK6ah-Pc8MybOHaCVjOW9fOJ7K8w 5208
modelopt/torch/sparsity/searcher.py sha256=nYr2zNcUHrzy5UxljnZGkGWioHYmlqTtIHYxRHMXWgk 3040
modelopt/torch/sparsity/mode.py sha256=YHqlKlPQqG_2Ez5wBnn7h0HLaSxc6rfuRV3F6RHbfos 6634
modelopt/torch/sparsity/module.py sha256=Ys0GXiifJYaL29ER_qSXD-AKBiVXvXFp0ze74oRkc8E 3504
modelopt/torch/sparsity/config.py sha256=eDeNgPICuPziqBilG0BIcOJNjbpzRvLiYuzY9bpQZRM 1593
modelopt/torch/sparsity/plugins/megatron.py sha256=pw_0-dq5qwYbxvOzKHMAIbb7Te_5XFBzqUpa_BGXzfs 2688
modelopt/torch/sparsity/plugins/__init__.py sha256=qgTlow9Zi5bja6DmZK08sUiELhg1uq8zBK22MWrxWws 813
modelopt/torch/_deploy/profiling.py sha256=FR4GOkk2O1YOxuh4C_10fY42KqOCDx8ayMq9Fg-VwNY 7400
modelopt/torch/_deploy/__init__.py sha256=go8A5rZnKiAJy-vxja_g7lM6lktTbyjHV6jLa9OThzw 858
modelopt/torch/_deploy/compilation.py sha256=0D0Nh8h1z-SRyrpAnNw_APYE3EcODm746Ze5tz-fXqA 5567
modelopt/torch/_deploy/device_model.py sha256=TLD6KR3GR3Ljprsvdp0fbMawlRojW7-ZkgZbMIR-bXU 6101
modelopt/torch/_deploy/utils/onnx_optimizer.py sha256=rmX4dFA_I6tQrQok2CHz94yOr4rKcHt7-xvzyDCNp2E 4619
modelopt/torch/_deploy/utils/__init__.py sha256=VTT1JZ8zKbDXprdb1MnAyNDKBMPyGQDHS1xVE6NIYPI 620
modelopt/torch/_deploy/utils/torch_onnx.py sha256=nQ56RZ2HDw8ZSD_MXNJxojSd29sX6Pir6pT4682nKEk 23149
modelopt/torch/_deploy/utils/onnx_utils.py sha256=j_sOc6GJTAElAbGSH5NUU0_ulQTyieVhTLhsSwD1U9k 2419
modelopt/torch/_deploy/_runtime/runtime_client.py sha256=tLlQN5jczmmGI7NxqJiQ7oFPzJRia8yQIyrUw4nAbf4 6399
modelopt/torch/_deploy/_runtime/__init__.py sha256=huZPwNKIjpGnMqNdLqSAMuultdzsiEHGR8GFUy7wTag 909
modelopt/torch/_deploy/_runtime/trt_client.py sha256=-65FgYAm6VRuZSWem9QUskvPLDLIQXrlFJINUBB7BZY 9183
modelopt/torch/_deploy/_runtime/registry.py sha256=pYjOXvVT-SC5aJzc8qyC93KGmK1kTijIVhcNN4sXvJo 2593
modelopt/torch/_deploy/_runtime/common.py sha256=S4l5FQB7BUWQtk5t3XlgELicsBogBUx4V9hvn00pZNQ 2259
modelopt/torch/_deploy/_runtime/ort_client.py sha256=kqYvpoui1g9Lt8RkVgX9USgde-TTH6yF4HrkwXyUAjs 7405
modelopt/torch/_deploy/_runtime/tensorrt/engine_builder.py sha256=XhwdCMii05IgKF3uBs6R3CmJKlt9nngjL7OUKBDWJ7A 12012
modelopt/torch/_deploy/_runtime/tensorrt/constants.py sha256=DHHutC6wtnaJc9a41DLT5dKjh_XKEYFG-qVfwswKlO0 2764
modelopt/torch/_deploy/_runtime/tensorrt/hw_param_config.py sha256=gpyL4Aei-xf5gTQr3KJJOJdFl8Y-wMaYF236ScUttRo 1719
modelopt/torch/_deploy/_runtime/tensorrt/layerwise_profiling.py sha256=RwpzibSW0yfPZUYHuF4l8FxHmGDoDEGQu1aSp5iEdrE 6775
modelopt/torch/_deploy/_runtime/tensorrt/parse_trtexec_log.py sha256=kzIq57RnM-54uZ_bB9d7QwvyzEhojY71fVynN67NMmY 5536
modelopt/torch/_deploy/_runtime/tensorrt/tensorrt_utils.py sha256=TYTTj0yyYqNLrPPMkE0sTS9gWLAmEd0kXE0LUx_hzQA 6320
modelopt/torch/export/model_config_utils.py sha256=RwfuHmLZVYZbD7Zmt-j_eBfX6zAUHduam9sKUdwSexQ 19146
modelopt/torch/export/hf_config_map.py sha256=a6cTZ3m0ZMCPwIdzJD_cqX4lMtELJKxxrLuW4J2IzOU 3965
modelopt/torch/export/__init__.py sha256=twkZUXt18Ak0bKzQ_5tRVPy0hFtAPTS9v0PJoqJolvY 734
modelopt/torch/export/tensorrt_llm_utils.py sha256=iQOveZWUxqZV_Lw9DnbOYw01s19q47_nmwEg2GuyORw 22886
modelopt/torch/export/layer_utils.py sha256=-3FZLlvyJWkCaY906wahLKTgtFVazEEIEjiIAppMHc8 66883
modelopt/torch/export/quantization_utils.py sha256=zI1AF8FYfDi53zX_bhMujCcNkyu8aNIkl-3YQKqCChA 29197
modelopt/torch/export/transformer_engine.py sha256=SQa6hoM5fNll5xXivvwMD2gtSXck6XIlNzT-ek5A80Y 4043
modelopt/torch/export/postprocess.py sha256=jMvrKahghSUfs6rNCL--PFZEk7EH59vmWwJojbt-G4A 32092
modelopt/torch/export/distribute.py sha256=yt8rthfPFQ-0LhRiXXirWO4AG005ZdLggTMKZW7WHlo 11382
modelopt/torch/export/model_config.py sha256=DSh_G7uols1Q6iRUugmMODO_Y65LYTMG0uBMJ9RTgEM 18591
modelopt/torch/export/tensorrt_llm_type.py sha256=W-Qp6u3XYOujLOHz7Aidz_dWZCWjJeuow5cQjGl5eWE 1171
modelopt/torch/export/model_config_export.py sha256=NeAQvJ69YwV4IVC3w-BWNHWnmqTyGCmobfRrdsDpHCc 25166
modelopt/torch/export/unified_export_hf.py sha256=63HJ1D2f1feyG7g1kTcyt6rRbnpHd2y1z_OcYSmuOrE 14825
modelopt/torch/export/mcore_config_map.py sha256=EvCpA7M6RuDahMRl9oqHMhZ5qObqrZPVbBTUAuy2Jhs 900
modelopt/torch/distill/distillation_model.py sha256=VdI0x81JExjwVVCPnRDI4t4Bb7UcjN5CRW7QZlYFXXs 11330
modelopt/torch/distill/distillation.py sha256=qKe0zWSBNXF7IBvX2dJQtXa40eX-KD_GfK6DTZHaGXc 3175
modelopt/torch/distill/__init__.py sha256=AMBXcfZ5TZIiRUpoxji1SomnS03lWrEFFvd00BcRsN4 858
modelopt/torch/distill/loss_balancers.py sha256=qQbUO3ANZH-PFEzTh0IkR_GVqFt0vlK94hKI8h4tDDE 5448
modelopt/torch/distill/registry.py sha256=NvUs7ocjVCqdqmxOKbbSeYB_WuavZmewEzJGLlODBk4 859
modelopt/torch/distill/mode.py sha256=D0W4yYfHnd4YYA03nvpxT46eYmgqf636teeQlkSoPDA 7963
modelopt/torch/distill/losses.py sha256=Xgxbmv5XtjOKfKWS1gOtWRiePfm1FGuDTr-dBLvyAUk 4986
modelopt/torch/distill/config.py sha256=tq0FocohobVtzDDBmGNwQfQ_ARM-P4F6dk8mj8faa78 4988
modelopt/torch/distill/plugins/megatron.py sha256=UsPDNbAGRihRz_5FnkQiZX2ldVTdgsDEWh2pdyWtP3o 18153
modelopt/torch/distill/plugins/__init__.py sha256=eAkvr2O3LIURNdE6Jeduy47yCwD-fW3VU0rrNH0PwfQ 717
modelopt/torch/opt/_hooks.py sha256=7w4TLRiyKa7yJXC5OGDEsABCZYa_JMVOllc9yXxN5bA 3074
modelopt/torch/opt/dynamic.py sha256=Y4nV73LvlkeQdz7F2tKRIiSYbo_s3tlv6Ijngmtvmfw 57952
modelopt/torch/opt/__init__.py sha256=cKrn2sUL9dDm7s2JExtrjYn5kMEaY8bO1x6jzKMRDFg 1381
modelopt/torch/opt/conversion.py sha256=P1SxsTv5SlP0DcNsewaubfDY0TjGeoDbXvVFwhnE5OE 24298
modelopt/torch/opt/searcher.py sha256=ThVBlEimgiGcy_XpYoenZSFOrii1_H24oF20zhLPuwY 14028
modelopt/torch/opt/hparam.py sha256=jKZxgsxXwE7D_BchoGpQZ2spBDJTnvE0MaSslolUuJo 10202
modelopt/torch/opt/mode.py sha256=qT5EFdAhKGJH7IE5k29_uB7LklxMpNvTDNm1c5ozvOk 12361
modelopt/torch/opt/utils.py sha256=jvwLAdcQQGcIlyJCa-SafqsXWO9bICE1rpJ8sfbRhTw 2482
modelopt/torch/opt/config.py sha256=9zcIkw2geUgR4y0KjJ7etpd1PUyi6wSJdp1x8PBjOzg 14834
modelopt/torch/opt/plugins/__init__.py sha256=_8eHaZwjZXXienVvOlbmPHBrwwjAKwUsBVN80A1VD8o 958
modelopt/torch/opt/plugins/peft.py sha256=-WQABxIcW6GGqa_ZxwWwcW1T4Chr6wa-JJAfqS1iXJM 3977
modelopt/torch/opt/plugins/transformers.py sha256=7wo2ZW29GHBPMxDmqrJyrnD7eurGA9U-G8IG07kqSOA 814
modelopt/torch/opt/plugins/mcore_dist_checkpointing.py sha256=BQ6BLeTxZIxbMq-SnXssbhf3DTe3igmOe24F70Tez0s 16122
modelopt/torch/opt/plugins/diffusers.py sha256=Brdn-DoUbDslR-KVM1svT-C-aMidn7VlLtfJxTCcMi0 799
modelopt/torch/opt/plugins/huggingface.py sha256=khlefLN3RuC6IlpNEpmE8gzvm8FQjijEwFeM01Zq-Hg 6541
modelopt/torch/trace/__init__.py sha256=1ddbXvCOHxuSrPopUDsmR7vTLmsGLgRZfWH_Z-4JGjk 812
modelopt/torch/trace/symbols.py sha256=uaRtY6CQJH_7K3IhqNpqWmQuinpi0wsBdXqGggMTGk0 21478
modelopt/torch/trace/analyzer.py sha256=lljvay7kfH1i83xBLBKqTM7VGUiUKdIvfiFGKvOUIEI 62930
modelopt/torch/trace/tracer.py sha256=XY4Pxmb5ulg3T0FMK5Tdu-xwvrif5pTd9aafG99Hdak 13115
modelopt/torch/trace/modules/nn.py sha256=jnnbC00UB4XkBQl9RyzPJwO_2JXsQDyfqF8GlgSBzvU 5563
modelopt/torch/trace/modules/concat.py sha256=iPBs3fNCnBsfUiR7YcyERrLYr_uZde30ognhdAucKg0 17392
modelopt/torch/trace/modules/__init__.py sha256=jg3ipg86bovfxy3dvA0oLDX5ZiPEXAdkjhEdWUzBr_w 671
modelopt/torch/trace/plugins/megatron.py sha256=7sgQSmChI-CeM1_G42kXY7i-T8xeNFsynrlR8PZ6y5o 1153
modelopt/torch/trace/plugins/__init__.py sha256=7XXD-BnO1gzfFgdomO3gLZpiEb1h9ZdZ5khUpNpEFkk 964
modelopt/torch/trace/plugins/transformers.py sha256=1Q4NfdFK9oDxdL3IIxJMU7_HP7EqFdjh_iTv_Ko-XxM 2119
modelopt/torch/nas/autonas.py sha256=8RSWu_D5Vk4VLqA3VLHEZIKBFCOC6I5EHvq7hJDO5T8 24371
modelopt/torch/nas/__init__.py sha256=2I2WsDgXqOCJVtdJtme-2DC7X-CUVfs07Ua6_xNAOXI 897
modelopt/torch/nas/_algorithms.py sha256=2MIdfRrkP-Bh0fWvwVmzN_faYU46jaYrN0ihi5I79_o 17580
modelopt/torch/nas/_patch.py sha256=vgk7ywtfJho8y_unk5lJQr6ASIQS9qb6vkVvKNto4AQ 10125
modelopt/torch/nas/conversion.py sha256=mq9vCyx8xph0flrV2W1T-Y7aX3nUtihLOOzmspC1ZDM 5287
modelopt/torch/nas/search_space.py sha256=zQORUsi4q9sqib75m0xuvh0cNh0meg3ovIpakLISx_c 10522
modelopt/torch/nas/algorithms.py sha256=t30Y0DLhblYF85JQU5Rq24rXze6Z-lCqgsVAWCjJOIU 19128
modelopt/torch/nas/registry.py sha256=agxYGizCCnfnzVpuBAz5XR-HKcWo6f0qU4eFZfqX7qA 809
modelopt/torch/nas/mode.py sha256=U8njUnGm6UkdRyuO4kNlAjTksFyQj9QrtWRxMjypO3I 4137
modelopt/torch/nas/traced_hp.py sha256=eZwDC2CsE3kf_1IInlOSd1yg344Is-OcpFsvV8ChVWk 5519
modelopt/torch/nas/utils.py sha256=12vp3NDyTgHOmOI6LgHp3uZBL3oYDqM8gPBxwEmez5g 13353
modelopt/torch/nas/config.py sha256=zCwQc_yi8gXWtZCkJOgk0-z9L1eoeDY_C901XNzj-qg 2808
modelopt/torch/nas/modules/__init__.py sha256=EkACv8qNF-556naXM0a-n0S1HOu2Qyzt0TIw_X1RShw 689
modelopt/torch/nas/modules/norm.py sha256=OP2vZK0jXA4swxU8wKrqLJOnOpQSOv7NuwVA2GfBYmk 7458
modelopt/torch/nas/modules/container.py sha256=3oy7_O5dK-8q_WyPD46VKT3RN7MG1rhUWIyG0GX1k3E 3363
modelopt/torch/nas/modules/utils.py sha256=zr9_fGJtKjyU_p5rF0nOrU2-Jp-vzi7SUhmAAJthd9U 2051
modelopt/torch/nas/modules/conv.py sha256=OPS9F2cccE0kto7zdpC2u1U4f3gB18Evzg4ub-Y5H4E 10505
modelopt/torch/nas/modules/linear.py sha256=FCUm3JYHlzWypgTzEx3hRXCq51dOWKDPx9qEboL94uY 3076
modelopt/torch/nas/plugins/__init__.py sha256=5vNvKmJdz9fg-dokGUED7o6i2-3lwxOV-0hQKijKLwI 913
modelopt/torch/nas/plugins/torch.py sha256=ZW2oYHjcoxS1EdFGJcXH84eUxP_PVh7eh8wRPtD2H9I 2442
modelopt/torch/nas/plugins/transformer_engine.py sha256=tAqlyV-CF9b70vUBgXw4pSgHnVZJkB4z1O3nktEz6ZY 1501
modelopt/torch/nas/plugins/transformers.py sha256=3JItlfA70qh6WN1y1yWdjq2UgvNXEaV8SucGnlBhXkQ 8178
modelopt/torch/nas/hparams/concat.py sha256=y6MB7LzeGpQdh16a6kfExoUJGP08RdZFuyVeD1qb8GA 12429
modelopt/torch/nas/hparams/__init__.py sha256=4ymvKGxPcXMg4yCEvlCJ5mq-OdMtVnUocLiJ0r7YAJs 675
modelopt/torch/nas/hparams/container.py sha256=2nCvmUYuqV1f9qCzUF95uRD-7jK45Vdh-FghFgCI0lw 1611
modelopt/torch/prune/mcore_gpt_minitron.py sha256=YKkiRDOPqi2ZOCxADNCI4SM2WIIcy93zz6jMX9XrIB4 6071
modelopt/torch/prune/__init__.py sha256=ZGH3rNm_5hBT7dioRcA9OA3KDWON6UGhQe7-BRa3GkM 978
modelopt/torch/prune/fastnas.py sha256=0rXpxhPMyeZ-ty7OfjtsEW0owJEs1hMSnR3ZSo4dU_I 10622
modelopt/torch/prune/gradnas.py sha256=mOe0dTRj0yBHyCj2bql7Wt46Ogwd6o3BP0RHq0y45Jk 12379
modelopt/torch/prune/mode.py sha256=8W2P6pWGHhYK44Hcc8jcE8Z9z7FQrB6L621MaqPVg3U 4690
modelopt/torch/prune/config.py sha256=XwobNF_yY7gV7e6UnKffvnabLnsXAcmBKVxO2X67W0w 2799
modelopt/torch/prune/pruning.py sha256=32T9j--Qa8rq-k1LRiP4wL6XaEk6cyLwqVAPkAU4q98 10533
modelopt/torch/prune/plugins/megatron.py sha256=se1Dq-AYVQ3hSOSH-Wt5WDeUWCUFng8p0us7C0Ly2kQ 1105
modelopt/torch/prune/plugins/__init__.py sha256=FSbqA05beLni1wr8wDP0QzehFL-KwNxxu_jLRuxii1Q 780
modelopt/torch/prune/plugins/transformers.py sha256=8z5mf17DsP-MZmd1OvPDR1ZthT4EuQXy1s8otR7e5-U 1159
modelopt/torch/speculative/speculative_decoding.py sha256=0ihEn5dquvVFgE63v59xODrIJrTNbfXsqTpZEDt_fAc 2425
modelopt/torch/speculative/__init__.py sha256=uSYsseVCc39UbqWC5KA-gecUWOtV5-frTsPy-Ilj4IY 692
modelopt/torch/speculative/mode.py sha256=RA9Z8BV12urVQN6AeacgEbu2cUesyrvfPqFgsM0EQAM 3659
modelopt/torch/speculative/utils.py sha256=9if6850ikl_ItpGwSD1UxPSfe3f3ufRV8HbQcC_uJM8 1671
modelopt/torch/speculative/config.py sha256=-_FkPoSJ5zdDjA4b6d5g7Jo-JSGZYlsNGhvhz7vDXK8 1594
modelopt/torch/speculative/redrafter/__init__.py sha256=gtQRIA1jp876DL_98zBi4v6TgviWLKvb8hnVdpzR9Ds 643
modelopt/torch/speculative/redrafter/conversion.py sha256=m3SvbVDGzduri5vRqyF94vy0ZOJJUnR1JE7sg64tVPw 2227
modelopt/torch/speculative/redrafter/redrafter_model.py sha256=UkbiYu-_dAeE_af09GQiJxjHkCX1_lO75c0S1txplpE 1238
modelopt/torch/speculative/medusa/__init__.py sha256=76HwQPCvR4tv5Es-OvnbUlgbaqGAJRu1Xkglw-esLBE 637
modelopt/torch/speculative/medusa/conversion.py sha256=GY4QdAz3XiAH9KB7_SK-NPmHqwCWa8oqW7MwYx6OhTc 2097
modelopt/torch/speculative/medusa/medusa_model.py sha256=VkALH1IwNQ3p9cmGfXGDeJCMGrwf8p4UjCR2gdHylUo 1176
modelopt/torch/speculative/plugins/megatron.py sha256=rbpovj3Axdxcwb80nwRrE6fduPJS34pqdDWECzB1QPo 27401
modelopt/torch/speculative/plugins/__init__.py sha256=Pp8ZbepOp1zNYSYFELQVmdmD_40WiS0LHrqCl73yxhU 1011
modelopt/torch/speculative/plugins/transformers.py sha256=2AvUToYfVgR-B3Q6nQTqwzfIB5zYmlbdDjzsU0-Nv-I 24929
modelopt/torch/speculative/eagle/__init__.py sha256=AY5Ra41DPEsGwY_f7Jui6hvzXXieoUVSrNRhkpXxGEk 635
modelopt/torch/speculative/eagle/eagle_model.py sha256=SLrmC0lK37l06sFO5wKLdpMXYO3DLZeu5sQuDZEtazY 1035
modelopt/torch/speculative/eagle/conversion.py sha256=UdeRA-z9xSNHXRp_epp60MAtPxzYlz0iw28UX8lqDUw 2014
modelopt/torch/speculative/eagle/utils.py sha256=ciukBIzqioyOaEmTh7U35nkwzJP9GqhRcJg_7bPQgu0 15210
modelopt/core/__init__.py sha256=YTwmZQUJhaDBjVUKHK1DMFcurwAcpXBX9ASe7YDsMWI 730
modelopt/core/torch/quantization/algorithms.cpython-38-x86_64-linux-gnu.so sha256=SCYD4rvc0HxpXmPQB2ZJm-cFrGx6NLT7XW8RGunaHBg 3077120
modelopt/core/torch/nas/plugins/megatron.cpython-38-x86_64-linux-gnu.so sha256=mMdOA2PqjmHUebMGUHVK2MW0_aDbbl9T2CQ-ycLI8QY 4474224
modelopt/onnx/__init__.py sha256=bNYgfPF0BShCfTNzLjreSucNGFNFzZXgFXjp9xVONuo 752
modelopt/onnx/op_types.py sha256=ldYF4aXeDOV-7NHYzbtx55W6GXBMFRmPHbEFMwmTSnc 7233
modelopt/onnx/utils.py sha256=FMZz2_Lg8nkAjalNietrhBGz0z4DHCG_toit2UxzbPM 20375
modelopt/onnx/quantization/ort_patching.py sha256=p8zaRssRAxPZeKxiWU74bhXLAw9FobJ7y11lkEB_Uzk 25112
modelopt/onnx/quantization/quant_utils.py sha256=NTaqmmLqBxB4p2XHMcHz2CFd4EOeCmrk8zfxUqnNJJE 6537
modelopt/onnx/quantization/__init__.py sha256=gKfOvG3E2_kAeW1sw6E9RIRhvoyTnuioWjfwLtRshBs 741
modelopt/onnx/quantization/graph_utils.py sha256=1DjFjsrfldoB_evWQ6YnWJfNg2agjVqIsfib1RGeQFk 40349
modelopt/onnx/quantization/ort_utils.py sha256=Kh2oahqk7jhKIWzwS_IYeRlMM4fU8OeWXCFAl7w-eaQ 5112
modelopt/onnx/quantization/fp8.py sha256=WjvYeRtM5hU9vTsaGsjoimkCGVd95rY62Pp5rFy3G9c 10760
modelopt/onnx/quantization/int8.py sha256=Jiw6ISPM9kT6Uc4d424B5vVnK8Q_pLgkj28yoMrZmtg 10361
modelopt/onnx/quantization/extensions.py sha256=Ah2rAyuLHraJ9-4GUMm5fK11Z__qLAl_HjDZZWQ0BZs 1080
modelopt/onnx/quantization/trt_utils.py sha256=-DE1qjWlVbJ1znga_-QlB4ABnoyhq8l0wCm_NEcjwAs 3983
modelopt/onnx/quantization/quantize.py sha256=0YlC2gi68HaJw_Uq5yI_7TbNsc1KQWJaDnvYRDmjGSs 17318
modelopt/onnx/quantization/gs_patching.py sha256=FhoF6T2Z3fwVbXJ2MxUus5hw5MtAfO5I819Uik0galI 4017
modelopt/onnx/quantization/partitioning.py sha256=NLCFzWtqHk1TqD0W25M_7py8936Kd95htcrzKjDZs30 14947
modelopt/onnx/quantization/int4.py sha256=Smnr9hZPkJa9FMbLggHedFANtTglEmjpctRI10tNk0A 50117
modelopt/onnx/quantization/__main__.py sha256=tB9h7fBv-uIt4R6jB9vLFXsNSgqtBNyNcdUIqoZEKpc 8157
modelopt/onnx/quantization/qdq_utils.py sha256=l_nIeI_H7Kpw5dwkkuIXRYBkHPnqFHpelKOjF5Jd4cs 28716
modelopt/onnx/quantization/calib_utils.py sha256=bP2tt1rK9vIwrAdS8VDchZ7CPt9hdnWWdhlxbzBMADA 5160
modelopt/onnx/quantization/operators.py sha256=Y_cQSdnSoP0h4LHpV1rK4-qISTgUMOtrv2YjFX1PTbk 4251
modelopt/onnx/quantization/src/modelopt_round_and_pack_ext.cpp sha256=YZA8R9RMNdqrhbhmWMBROHb1sjm5KPId37x_5GBJFuY 4088
modelopt/deploy/__init__.py sha256=duhegEd1OSrG5RzIhV1raqtwyS-qJlHX28cp1JOa7R8 593
modelopt/deploy/llm/__init__.py sha256=0jzH15h7-ATGynklKbtngl114Y_95HaWW5Cj-vKu9hI 1031
modelopt/deploy/llm/nemo_utils.py sha256=Xs9RofolZ3GBhp0-Tsmmszypg7MnX7sNvsFpQPcmPnk 7158
modelopt/deploy/llm/generate.py sha256=mm5SJlR1QqFA1RQ6TTC-WEdHfgBJfJz0-Ba_xJ14Ru4 9930
nvidia_modelopt-0.21.0.dist-info/top_level.txt sha256=IMo1VR4nTJCILUd5BAb4Pd5DmXMRn_4W9wobCeTfn5Y 9
nvidia_modelopt-0.21.0.dist-info/WHEEL sha256=Kn109NXuT8KgFUFBR7C2ZbCThqiQvDDPWZsf0xScqBs 147
nvidia_modelopt-0.21.0.dist-info/RECORD
nvidia_modelopt-0.21.0.dist-info/LICENSE sha256=Il5n7fbhbOotbSs3WkWaBPXy6-Sa8pYiPxnZSb3CJ9E 531
nvidia_modelopt-0.21.0.dist-info/METADATA sha256=ULoY2lTWvoNKeXp-mEEcVE4TkoKw-h2lyv2QhRAoVRE 3549

top_level.txt

modelopt