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])