Model Export & Deployment (jlnn.export)

The export module provides tools for transforming trained JLNN models into formats suitable for production deployment. Thanks to JAX, we support native export to standardized formats.

• Metadata & State Export
  • export_module_recursive()
  • extract_logic_parameters()
• ONNX Export
  • export_to_onnx()
  • export_to_stablehlo()
  • export_workflow_example()
  • save_for_xla_runtime()
  • Implementation
• StableHLO Integration
  • export_to_stablehlo()
  • export_workflow()
  • inspect_stablehlo_module()
  • save_stablehlo_artifact()
• PyTorch Mapping
  • export_to_pytorch()
  • verify_pytorch_conversion()
  • Key Features: