Metadata-Version: 2.1
Name: nsight-pytorch2xdl
Version: 0.0.2
Summary: Exporter library for PyTorch to NVIDIA Nsight Deep Learning Designer XDL format
Author-email: NVIDIA Corporation <nsight-dl-feedback@nvidia.com>
Classifier: Development Status :: 4 - Beta
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Intended Audience :: Developers
Classifier: Programming Language :: Python
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Requires-Python: >=3.11
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: torch >=2.10.0
Requires-Dist: numpy
Requires-Dist: coloredlogs
Requires-Dist: verboselogs
Provides-Extra: doc
Requires-Dist: sphinx ; extra == 'doc'
Requires-Dist: nvidia-sphinx-theme ; extra == 'doc'
Requires-Dist: sphinx-autodoc-typehints ; extra == 'doc'
Requires-Dist: furo ; extra == 'doc'
Requires-Dist: sphinx-mdinclude ; extra == 'doc'
Provides-Extra: test
Requires-Dist: pytest >=8.0 ; extra == 'test'
Requires-Dist: torchvision ; extra == 'test'

# Nsight PyTorch2XDL

Nsight PyTorch2XDL is a Python library that provides tools for exporting PyTorch models to [NVIDIA Nsight Deep Learning Designer](https://developer.nvidia.com/nsight-dl-designer) XDL format.
The library handles the conversion of PyTorch operators to their corresponding XDL operators. It also provides functionality to export the model's weights.

## Installing from PyPI
Nsight PyTorch2XDL can be installed directly from the Python Package Index (PyPI).

### Basic Installation
For most users, a basic installation is sufficient:

```
pip install nsight-pytorch2xdl
```

## Installation from source

Install as an editable install:

```bash
pip install -e .
```

If you want to manage all run-time dependencies yourself, also pass the `--no-deps` flag.

## Package Dependencies

The required package dependencies can be installed with

```
python -m pip install -r requirements.txt
```

## License

All files in this package are subject to the Apache 2.0 license.

## Disclaimer

nsight-pytorch2xdl is in a Beta state. Beta products may not be fully functional, may contain errors or design flaws, and may be changed at any time without notice. We appreciate your feedback to improve and iterate on our Beta products.
