google/jaxonnxruntime issues and pull requests

#145 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] 12 days ago

#144 - Add `lowering_platforms` to `traced.lower()`

Pull Request - State: closed - Opened by copybara-service[bot] 23 days ago

#143 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] about 1 month ago

#142 - Replace `XLACompatibleSharding` with `jax.sharding.Sharding` since the former is deprecated and will be removed in the future.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 month ago

#141 - Replace `XLACompatibleSharding` with `jax.sharding.Sharding` since the former is deprecated and will be removed in the future.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 month ago

#140 - Replace deprecated `jax.tree_` functions with `jax.tree.`

Pull Request - State: closed - Opened by copybara-service[bot] about 2 months ago

#139 - Bump minimum jaxlib vesrion to 0.4.27

Pull Request - State: open - Opened by copybara-service[bot] 2 months ago

#108 - Create the tutorial of call_torch API

Pull Request - State: open - Opened by maxwillzq 11 months ago

#101 - Add test_addictive_attention call_torch test.

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#100 - Add more python tests.

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#99 - Start to add more pytorch test case base on https://d2l.ai/.

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#98 - Add onnx op Not

Issue - State: closed - Opened by maxwillzq 11 months ago - 1 comment

#97 - Change call_torch API format from `f(inputs)` to f(params, inputs).

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#96 - To be consistent with onnx, exclude those ops not in the old model opset.

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#95 - add vlog for jax device.

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#94 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#93 - improve the call_onnx_test.

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#92 - Fix bugs to pass more onnx backend tests.

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#91 - Enable the call_torch experimental API.

Pull Request - State: closed - Opened by copybara-service[bot] 12 months ago - 2 comments

#89 - Modify README.md to encourage open source contribution.

Pull Request - State: closed - Opened by copybara-service[bot] 12 months ago - 1 comment

#88 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] 12 months ago

#87 - Add supports_device to backend module members.

Pull Request - State: closed - Opened by copybara-service[bot] 12 months ago

#86 - Add doc for adding a new op.

Pull Request - State: closed - Opened by copybara-service[bot] 12 months ago - 1 comment

#85 - Add ops to support ProteinMPNN.

Pull Request - State: closed - Opened by copybara-service[bot] 12 months ago - 1 comment

#84 - Cleanup Copyrights

Pull Request - State: closed - Opened by zhiqwang 12 months ago - 4 comments
Labels: pull ready

#83 - Implement more ONNX Ops.

Pull Request - State: closed - Opened by copybara-service[bot] 12 months ago - 1 comment

#82 - Update the onnx op code generator.

Pull Request - State: closed - Opened by copybara-service[bot] 12 months ago - 1 comment

#81 - cast op from_type must get value from real input tensor

Issue - State: closed - Opened by maxwillzq 12 months ago

#80 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] 12 months ago - 1 comment

#79 - Improve the context_graph.initializer_dict as dict[str, jax.Array].

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#78 - Use `jax.eval_shape` API to do shape inference. So there is no need real input when tracing the JAX function.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#77 - Change py_test do have typing check.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#76 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago

#75 - Move global config to main file.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#74 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#73 - Preprocess to format all tensor names within an `onnx.GraphProto`.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#72 - Update issue templates

Pull Request - State: closed - Opened by maxwillzq about 1 year ago - 1 comment
Labels: pull ready

#71 - Demo how to use tf create onnx custom op.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#70 - Update build.yml to run tests under onnx==1.14.0

Pull Request - State: closed - Opened by jinning-li about 1 year ago - 2 comments
Labels: pull ready

#69 - Add implementation of older op versions.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#68 - Fix operators to pass onnx==1.14.0 op tests.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#67 - Fix python typing of Cast op.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#66 - Create transformerblock custom onnx op to demo how to create custom op

Issue - State: closed - Opened by maxwillzq about 1 year ago

#65 - Update README.md and contributing.md.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#64 - Update version to 0.1.0.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#63 - Update documentations.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#62 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#61 - Add older version of operators.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago

#60 - Ignore np.object (str) test since it is not valid jax.array.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#59 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#58 - Add implementation of lower versions of the operators.

Pull Request - State: closed - Opened by jinning-li about 1 year ago - 1 comment
Labels: pull ready

#57 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#56 - OnnxBackendRealModelTest::test_squeezenet_cpu fails

Issue - State: closed - Opened by maxwillzq about 1 year ago - 2 comments

#55 - fix onnx 1.14.0 averagepool and cast bug

Pull Request - State: closed - Opened by maxwillzq about 1 year ago - 1 comment
Labels: pull ready

#54 - Update contributing md file.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#53 - Add onnx_op higher version shortcut for onnx 1.14.0 test suites

Pull Request - State: closed - Opened by maxwillzq about 1 year ago - 1 comment
Labels: pull ready

#52 - Revert "Add onnx_op higher version shortcut for onnx 1.14.0 test suites"

Pull Request - State: closed - Opened by maxwillzq about 1 year ago

#51 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#50 - Add onnx_op higher version shortcut for onnx 1.14.0 test suites

Pull Request - State: closed - Opened by maxwillzq about 1 year ago - 3 comments
Labels: pull ready

#49 - Rewrite the Max op to be compatible with the onnx Max op definition.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#48 - Implement the `If` op.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#47 - Add necessary mechanisms to deal with `tensor_dict` when subgraph presents.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#46 - Refactor the `call_onnx` function.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#45 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#44 - Add support for operators required by LLaMA.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#43 - Run full onnx test suite and fix those bugs as more as possible.

Issue - State: open - Opened by maxwillzq about 1 year ago - 4 comments

#42 - Improve the topologic sort algorithm.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#41 - Here we add config class and use it to control those workaround options.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#40 - fix init.py for pep 484.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#39 - Code clean up.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#38 - Remove input data dependency for call_onnx tracing

Issue - State: closed - Opened by maxwillzq about 1 year ago - 1 comment

#37 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago

#36 - Add model run-through tests for both internal and github users.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#35 - Test LLAMA model on jaxonnxruntime

Issue - State: closed - Opened by maxwillzq about 1 year ago - 1 comment

#34 - Internal.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#33 - fix the hardcode shape for eval_shape. Here we already cache the `download_model` function.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#32 - add size option for NonZero.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#31 - Add Local Response Normalization to support alexnet, googlenet, etc. Also fix minor problems with operator jit mode.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#30 - Add operators and fix scripts to support 'vgg 16', 'gpt-2', and 'bert-squad' models from onnx model zoo.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#29 - Cannot jax.jit the nonzero operator

Issue - State: closed - Opened by jinning-li about 1 year ago - 2 comments

#28 - Remove orbax in pyproject. orbax depend on tensorflow. It introduce a lot dependency.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago

#27 - Fix a typo on github workflow.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#26 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago

#25 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 2 comments

#24 - Run through the onnx model zoo gpt model and add those missing ops

Issue - State: closed - Opened by maxwillzq about 1 year ago - 1 comment

#23 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 2 comments

#22 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#21 - Implement the leakyRelu onnx op

Issue - State: closed - Opened by maxwillzq about 1 year ago - 1 comment

#20 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 2 comments

#19 - add pythonpublish.yml

Pull Request - State: closed - Opened by maxwillzq about 1 year ago - 1 comment

#18 - Add the first resnet50 example main program.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#17 - Add all ops and fix bugs for resnet50 model run through.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#16 - Add those ops needed for resnet50.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#15 - Create tool script to analyze model for those not-implemented ops.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#14 - add analyze_model tool

Pull Request - State: closed - Opened by maxwillzq about 1 year ago

#13 - Add first example how to use it.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#12 - Internal change

Pull Request - State: open - Opened by copybara-service[bot] about 1 year ago - 1 comment

#11 - Bump onnx from 1.12.0 to 1.13.0

Pull Request - State: closed - Opened by dependabot[bot] about 1 year ago - 3 comments
Labels: dependencies

#10 - Add support GPT-2 model needed ops.

Pull Request - State: closed - Opened by maxwillzq about 1 year ago

#9 - Add those ops enable GPT-2 model.

Pull Request - State: closed - Opened by maxwillzq about 1 year ago - 1 comment

GitHub / google/jaxonnxruntime issues and pull requests