Metadata-Version: 2.4
Name: edge-mdt-tpc-nightly
Version: 1.1.0.20250407.80138
Summary: EdgeMDT TPC package
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Operating System :: OS Independent
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Description-Content-Type: text/markdown
License-File: LICENSE.md
Requires-Dist: model-compression-toolkit>=2.3
Dynamic: license-file

<div align="center" markdown="1">
<p>
      <a href="https://github.com/SonySemiconductorSolutions/aitrios-edge-mdt-tpc/" target="_blank">
        <img src="/docsrc/images/edgeMdtTpcHeader.png" width="1000"></a>
</p>
  
______________________________________________________________________


<p align="center">
  <a href="https://github.com/SonySemiconductorSolutions/aitrios-edge-mdt-tpc/releases"><img src="https://img.shields.io/github/v/release/SonySemiconductorSolutions/aitrios-edge-mdt-tpc" /></a>
  <a href="https://github.com/SonySemiconductorSolutions/aitrios-edge-mdt-tpc/blob/main/LICENSE.md"><img src="https://img.shields.io/badge/license-Apache%202.0-blue" /></a>
 </p>    
</div>

__________________________________________________________________________________________________________

## <div align="center">Getting Started</div>
### Quick Installation
To install the TPC package, run:
```
pip install edge-mdt-tpc 
```

### Using the TPC

To initialize a TPC and integrate it with MCT, use the `get_target_platform_capabilities` function as follows:

```python
from edgemdt_tpc import get_target_platform_capabilities
import model_compression_toolkit as mct

# Get a TPC object representing the imx500 hardware and use it for PyTorch model quantization in MCT
tpc = get_target_platform_capabilities(tpc_version='1.0', device_type='imx500')

# Apply MCT on your pre-trained model using the TPC
quantized_model, quantization_info = mct.ptq.pytorch_post_training_quantization(
    in_module=pretrained_model, # Replace with your pretrained model.
    representative_data_gen=dataset, # Replace with your representative dataset.
    target_resource_utilization=tpc)
```


## <div align="center">Supported Versions</div>

<details id="supported-versions">
  <summary>Supported Versions Table</summary>

|                       |                    TPC 1.0                                                                                                                                                                                                                                                                    |
|-----------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
| IMX500 Converter 3.14 | [![Run Tests](https://github.com/SonySemiconductorSolutions/aitrios-edge-mdt-tpc/actions/workflows/run_tests_conv314_tpc10.yml/badge.svg)](https://github.com/SonySemiconductorSolutions/aitrios-edge-mdt-tpc/actions/workflows/run_tests_conv314_tpc10.yml) |

</details>

## <div align="center">Target Platform Capabilities (TPC)</div>

### About 

TPC is our way of describing the hardware that will be used to run and infer with models that are optimized using the MCT.
The TPC includes different parameters that are relevant to the hardware during inference (e.g., number of bits used in some operator for its weights/activations, fusing patterns, etc.)

<div align="center" markdown="1">
<p>
      <a href="https://github.com/SonySemiconductorSolutions/aitrios-edge-mdt-tpc/" target="_blank">
        <img src="/docsrc/images/tpcArch.png" width="400"></a>
</p>
</div>

## <div align="center">License</div>
The EdgeMDT-TPC package is licensed under Apache License Version 2.0. By contributing to the project, you agree to the license and copyright terms therein and release your contribution under these terms.

<a href="https://github.com/SonySemiconductorSolutions/EdgeMDT-TPC/blob/main/LICENSE.md"><img src="https://img.shields.io/badge/license-Apache%202.0-blue" /></a>
