brian-parbhu / onnx-jax

onnx runtime based on jax

onnx-jax

onnx runtime based on jax

Quick start

import onnx
from onnx_jax.backend import run_model

onnx_model = onnx.load_model('/path/to/model.onnx')
outputs = run_model(onnx_model, [input_in_numpy])

Demo

demo.ipynb