Module python_integration 

Python Integration Module for ToRSh FX

This module provides comprehensive Python bindings and PyTorch interoperability for the ToRSh FX graph framework, enabling seamless integration with Python ML ecosystems.

Structs

DeploymentPackage

Deployment package structure

GeneratedPythonCode

Generated Python code structure

PyTorchModelMetadata

PyTorch model metadata

PythonBindingConfig

Python binding configuration

PythonCodeGenOptions

Python code generation options

PythonIntegrationService

Python integration service

TrainingInfo

Training metadata information

Enums

PythonDeploymentTarget

Python deployment target

PythonFramework

Supported Python ML frameworks

Functions

create_jax_integration

Create a JAX integration service

create_pytorch_integration

Convenience functions for Python integration Create a PyTorch integration service

generate_python_api

Generate Python bindings for a graph

graph_to_pytorch_code

Convert FxGraph to PyTorch model code