TorchLean API

NN.Runtime.JAX

JAX interop #

Import static Jaxpr computations into the existing TorchLean graph and payload APIs. The Python adapter is NN/Runtime/JAX/jaxpr.py; JAX is needed only for capture.