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.
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.