Nsight PyTorch2XDL

Nsight PyTorch2XDL is a Python library that provides tools for exporting PyTorch models to NVIDIA Nsight Deep Learning 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:

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.