NIRTorch API Documentation

This page lists functions and classes exposed by the nirtorch package for converting between PyTorch models and NIR.

Main Functions¶

• Graph extraction: Convert PyTorch nn.Module to NIR graph

• Graph loading: Convert NIR graph to PyTorch nn.Module

For detailed usage examples and API documentation, see:

• PyTorch developer guide

• nirtorch source code

Key Modules¶

• Tracing: Extract computational graphs from PyTorch models - see tracing guide

• Interpreting: Load NIR graphs as PyTorch modules - see interpreting guide

• State management: Handle stateful operations - see state guide