.gitignore
.pre-commit-config.yaml
LICENSE
README.md
contributing.md
pylintrc
pyproject.toml
.github/ISSUE_TEMPLATE/feature_request.yml
.github/workflows/build.yml
.github/workflows/pythonpublish.yml
docs/adding_a_new_op.rst
docs/conf.py
docs/index.rst
docs/requirements.txt
docs/_static/css/jaxonnxruntime_theme.css
docs/examples/jaxonnxruntime-python-api.py
examples/imagenet/imagenet_main.py
jaxonnxruntime/__init__.py
jaxonnxruntime/backend.py
jaxonnxruntime/call_onnx.py
jaxonnxruntime/config_class.py
jaxonnxruntime/runner.py
jaxonnxruntime/version.py
jaxonnxruntime.egg-info/PKG-INFO
jaxonnxruntime.egg-info/SOURCES.txt
jaxonnxruntime.egg-info/dependency_links.txt
jaxonnxruntime.egg-info/requires.txt
jaxonnxruntime.egg-info/top_level.txt
jaxonnxruntime/core/__init__.py
jaxonnxruntime/core/handler.py
jaxonnxruntime/core/onnx_graph.py
jaxonnxruntime/core/onnx_node.py
jaxonnxruntime/core/onnx_utils.py
jaxonnxruntime/experimental/call_torch/Notes.md
jaxonnxruntime/experimental/call_torch/__init__.py
jaxonnxruntime/experimental/call_torch/call_torch.py
jaxonnxruntime/experimental/call_torch/call_torch_test_util.py
jaxonnxruntime/experimental/call_torch/tests/call_torch_test.py
jaxonnxruntime/experimental/call_torch/tests/d2l_torch.py
jaxonnxruntime/experimental/custom_ops/__init__.py
jaxonnxruntime/experimental/custom_ops/zeros_like.py
jaxonnxruntime/experimental/custom_ops/zeros_like_test.py
jaxonnxruntime/onnx_ops/__init__.py
jaxonnxruntime/onnx_ops/abs.py
jaxonnxruntime/onnx_ops/acos.py
jaxonnxruntime/onnx_ops/acosh.py
jaxonnxruntime/onnx_ops/add.py
jaxonnxruntime/onnx_ops/and_op.py
jaxonnxruntime/onnx_ops/argmax.py
jaxonnxruntime/onnx_ops/argmin.py
jaxonnxruntime/onnx_ops/asin.py
jaxonnxruntime/onnx_ops/asinh.py
jaxonnxruntime/onnx_ops/atan.py
jaxonnxruntime/onnx_ops/atanh.py
jaxonnxruntime/onnx_ops/averagepool.py
jaxonnxruntime/onnx_ops/batchnormalization.py
jaxonnxruntime/onnx_ops/bitshift.py
jaxonnxruntime/onnx_ops/cast.py
jaxonnxruntime/onnx_ops/castlike.py
jaxonnxruntime/onnx_ops/ceil.py
jaxonnxruntime/onnx_ops/clip.py
jaxonnxruntime/onnx_ops/concat.py
jaxonnxruntime/onnx_ops/constant.py
jaxonnxruntime/onnx_ops/constantofshape.py
jaxonnxruntime/onnx_ops/conv.py
jaxonnxruntime/onnx_ops/cos.py
jaxonnxruntime/onnx_ops/cosh.py
jaxonnxruntime/onnx_ops/div.py
jaxonnxruntime/onnx_ops/dropout.py
jaxonnxruntime/onnx_ops/einsum.py
jaxonnxruntime/onnx_ops/equal.py
jaxonnxruntime/onnx_ops/erf.py
jaxonnxruntime/onnx_ops/exp.py
jaxonnxruntime/onnx_ops/expand.py
jaxonnxruntime/onnx_ops/flatten.py
jaxonnxruntime/onnx_ops/gather.py
jaxonnxruntime/onnx_ops/gatherelements.py
jaxonnxruntime/onnx_ops/gemm.py
jaxonnxruntime/onnx_ops/globalaveragepool.py
jaxonnxruntime/onnx_ops/identity.py
jaxonnxruntime/onnx_ops/if_op.py
jaxonnxruntime/onnx_ops/leakyrelu.py
jaxonnxruntime/onnx_ops/less.py
jaxonnxruntime/onnx_ops/lessorequal.py
jaxonnxruntime/onnx_ops/log.py
jaxonnxruntime/onnx_ops/logsoftmax.py
jaxonnxruntime/onnx_ops/lrn.py
jaxonnxruntime/onnx_ops/matmul.py
jaxonnxruntime/onnx_ops/max.py
jaxonnxruntime/onnx_ops/maxpool.py
jaxonnxruntime/onnx_ops/min.py
jaxonnxruntime/onnx_ops/mul.py
jaxonnxruntime/onnx_ops/neg.py
jaxonnxruntime/onnx_ops/nonzero.py
jaxonnxruntime/onnx_ops/onehot.py
jaxonnxruntime/onnx_ops/onnx_not.py
jaxonnxruntime/onnx_ops/or_op.py
jaxonnxruntime/onnx_ops/pad.py
jaxonnxruntime/onnx_ops/pow.py
jaxonnxruntime/onnx_ops/prelu.py
jaxonnxruntime/onnx_ops/range.py
jaxonnxruntime/onnx_ops/reciprocal.py
jaxonnxruntime/onnx_ops/reducemax.py
jaxonnxruntime/onnx_ops/reducemean.py
jaxonnxruntime/onnx_ops/reducesum.py
jaxonnxruntime/onnx_ops/relu.py
jaxonnxruntime/onnx_ops/reshape.py
jaxonnxruntime/onnx_ops/selu.py
jaxonnxruntime/onnx_ops/shape.py
jaxonnxruntime/onnx_ops/sigmoid.py
jaxonnxruntime/onnx_ops/sin.py
jaxonnxruntime/onnx_ops/sinh.py
jaxonnxruntime/onnx_ops/slice.py
jaxonnxruntime/onnx_ops/softmax.py
jaxonnxruntime/onnx_ops/softplus.py
jaxonnxruntime/onnx_ops/split.py
jaxonnxruntime/onnx_ops/sqrt.py
jaxonnxruntime/onnx_ops/squeeze.py
jaxonnxruntime/onnx_ops/sub.py
jaxonnxruntime/onnx_ops/sum.py
jaxonnxruntime/onnx_ops/tanh.py
jaxonnxruntime/onnx_ops/topk.py
jaxonnxruntime/onnx_ops/transpose.py
jaxonnxruntime/onnx_ops/trilu.py
jaxonnxruntime/onnx_ops/unsqueeze.py
jaxonnxruntime/onnx_ops/where.py
tests/call_onnx_test.py
tests/onnx_models_test.py
tests/onnx_ops_test.py
tests/run_all_tests.sh
tests/core/handler_test.py
tests/core/onnx_graph_test.py
tests/core/onnx_node_test.py
tests/core/onnx_utils_test.py
tests/onnx_ops/dropout_test.py
tests/onnx_ops/onehot_test.py
tools/analyze_model.py
tools/op_code_generator.py