PyTorch Round-Trip Driver

This module assembles the state-dict round-trip examples.

It does not re-implement model math. Instead it wires together:

• PyTorch exporters (MLP/Export, CNN/Export, Transformer/Export)
• typed PyTorch JSON importers (MLP/Import, CNN/Import, Transformer/Import)
• native tensor operations for the MLP forward pass
• public executable nn modules for the CNN and Transformer forward passes

Run via the TorchLean example runner:

lake exe torchlean pytorch_roundtrip --model mlp|cnn|transformer --action export|import

Design goals:

• keep paths/dimensions centralized (no duplicated constants across examples),
• keep the example output deterministic and readable,
• keep this example import-safe (no root-level main that collides with other executables).

CLI model/action selection

Paths and fixed example dimensions

Small IO helpers

Export actions

Import actions

def NN.Examples.Interop.PyTorch.Roundtrip.mlpOutput : IO (TorchLean.Tensor Float [1])

def NN.Examples.Interop.PyTorch.Roundtrip.cnnOutput : IO (TorchLean.Tensor Float [2])

def NN.Examples.Interop.PyTorch.Roundtrip.transformerOutput : IO (TorchLean.Tensor Float [1, 2])

Public entrypoint called from the examples runner

def NN.Examples.Interop.PyTorch.Roundtrip.main (args : List String) : IO Unit