jaxonnxruntime

View on PyPIReverse Dependencies (0)

0.3.0 jaxonnxruntime-0.3.0-py3-none-any.whl

Wheel Details

Project: jaxonnxruntime
Version: 0.3.0
Filename: jaxonnxruntime-0.3.0-py3-none-any.whl
Download: [link]
Size: 177663
MD5: 3fea9da1f7a0c15c5f3fc57d127b1c27
SHA256: 72814405d611d549c1a172cfff214c1e7cff5d2b3737c3129970630f8ea5e466
Uploaded: 2023-08-18 18:00:47 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: jaxonnxruntime
Version: 0.3.0
Summary: Jaxonnxruntime: JAX based ONNX Runtime.
Author-Email: Jaxonnxruntime team <jaxonnxruntime-dev[at]google.com>
Project-Url: homepage, https://github.com/google/jaxonnxruntime
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3.7
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Dist: numpy
Requires-Dist: jax
Requires-Dist: jaxlib
Requires-Dist: onnx (==1.12.0); extra == "all"
Requires-Dist: mypy; extra == "testing"
Requires-Dist: pytest; extra == "testing"
Requires-Dist: pytest-cov; extra == "testing"
Requires-Dist: pytest-custom-exit-code; extra == "testing"
Requires-Dist: pytest-xdist; extra == "testing"
Requires-Dist: absl-py; extra == "testing"
Provides-Extra: all
Provides-Extra: testing
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 2018 characters]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.41.1)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
jaxonnxruntime/__init__.py sha256=IB8Ra822HMJDzShdz81DnSFcM9BaTE2RNlE6xWHSwDI 824
jaxonnxruntime/backend.py sha256=Fp9RFKzwwt93PbIhKUYLju0cMzslzUoAeK0-lwowym0 5263
jaxonnxruntime/call_onnx.py sha256=-YINqGkMiQEEvYCmUMuLinTZZ-9IT9rRDZ6U2hI9qvM 8468
jaxonnxruntime/config_class.py sha256=IYjQnIakrmF9V-UbpgBKQ2U_Jry2ENuou8_blLWDpyY 15332
jaxonnxruntime/runner.py sha256=yO0t3ysK6lzhcxb2gb1h57ogVD1aDni8R4zSEmsVJjg 12924
jaxonnxruntime/version.py sha256=cTASmVd0-EKebgjsu5rQtuVigMr60lJ50yEZzVRYFFo 670
jaxonnxruntime/core/__init__.py sha256=iIW_9kcG11o9q_bBQ4lwEeVqatMMGMgqiSQeIUSRV8k 1183
jaxonnxruntime/core/handler.py sha256=Tg0iHDYYmsZZahUFxhzMyxb9mcXirK7VBNk-_nY5Teo 3691
jaxonnxruntime/core/onnx_graph.py sha256=sRQQMccreLeSxiqbsJf37_APurSFQe1jHloV2bdVT4Y 7324
jaxonnxruntime/core/onnx_node.py sha256=4nP-Ze4vgmvEUf5FxYU7Fx3ohF0bLRKdPqEScfWEGDY 4689
jaxonnxruntime/core/onnx_utils.py sha256=4okr3jlYv1SuJk3bCPhkrLmopYrTluUPZjSEaAib8hw 11832
jaxonnxruntime/experimental/call_torch/Notes.md sha256=IfOlV724SnIv-xT2d9HNcvZb9u3hs5ozU2hhi16pQgk 80
jaxonnxruntime/experimental/call_torch/__init__.py sha256=ToTnvXayP7ou_wB6YIT0koMI7Ef2Gd_8t8JvfzJvriU 803
jaxonnxruntime/experimental/call_torch/call_torch.py sha256=Z268AM50BsmHZk8wmtXAN0Q8Li1d5G2NTPy-v0hEhMY 2810
jaxonnxruntime/experimental/call_torch/call_torch_test_util.py sha256=AKJkldHTiSC2ru_wxgMrDG3Zkc2US0hoJxM5nddBpLU 1875
jaxonnxruntime/experimental/call_torch/tests/call_torch_test.py sha256=OZxR1s75h5hxN2sDG67e9RoRZN_cvTJWdv5dfhU3s7Q 4037
jaxonnxruntime/experimental/call_torch/tests/d2l_torch.py sha256=tGZcnW9o3uFnS8T8_o53vGOdW_ADQ6rVwMmztKNoKEE 131855
jaxonnxruntime/experimental/custom_ops/__init__.py sha256=0hFsCo3Fvj3Tx_ShU1PKk1xZViXR-2ytw5Jg3bTol2s 766
jaxonnxruntime/experimental/custom_ops/zeros_like.py sha256=hYfD2oqqhBRdQ64iQHVjKjpFASz8EUrNVM_wzd2uGXo 2745
jaxonnxruntime/experimental/custom_ops/zeros_like_test.py sha256=_L4gwDBuHVas9Xg72aZbCXb6PyeL9-R5lJehAa8TrR4 2310
jaxonnxruntime/onnx_ops/__init__.py sha256=jHWJVvDBShkEZN01mR9m3-v8oeQqfm-80Qil41gc_mw 3366
jaxonnxruntime/onnx_ops/abs.py sha256=RDq-DDLTV1WRZIxQze5n4mVezYN9OUZNtaUHzq73Vws 2772
jaxonnxruntime/onnx_ops/acos.py sha256=IRqa0bVq12G_CEGMZfa1MS-b1F2XUBtUcAnxj-PqkLw 1889
jaxonnxruntime/onnx_ops/acosh.py sha256=OFeLNziKc6PT9z-PU5T-pWPCxy4uGK_OhkHZczkv0X8 1899
jaxonnxruntime/onnx_ops/add.py sha256=ku1FOZBEEmQ6ixw2Iu-6RjOMm1u0w4_bmLkrgh477Wk 3203
jaxonnxruntime/onnx_ops/and_op.py sha256=dioZluArXPrNPd26FvJBwS7TRVSuccj23I83Gl-3LDE 1882
jaxonnxruntime/onnx_ops/argmax.py sha256=9eBUMZTvEZTBXoIEDIvnU5biPRwK7T4870k6WZe5Wgs 2155
jaxonnxruntime/onnx_ops/argmin.py sha256=0LDXt2Rc630Urs8G5wFlXNA4aW5KXlpj2Q44jYa9jJI 2045
jaxonnxruntime/onnx_ops/asin.py sha256=Af1LB8HfZU9ZvYeYJ8PyZmafNAC0ZhxjRloDehpNSAo 1889
jaxonnxruntime/onnx_ops/asinh.py sha256=YjDeqhytWqaGSNOYaWYcv7sSxGZm5Rum2fHM1zEFnvo 1899
jaxonnxruntime/onnx_ops/atan.py sha256=2JIQRxSOu_K7II60OIOYuSA7ZjxZ8dTIrAv2jiGfX1g 1889
jaxonnxruntime/onnx_ops/atanh.py sha256=BrXnIy2pqk_urkQ3Re5B83_LLXGhVnK2MSno0jFwa7A 1899
jaxonnxruntime/onnx_ops/averagepool.py sha256=COO3S8BgxPhxLXu2GKVH9RB6xSXSwOVl5pEs-TrskQQ 3561
jaxonnxruntime/onnx_ops/batchnormalization.py sha256=YioCF02Xy6WyKfll5utEZLzeZR1TVFwl1xT-4BHrXxA 3486
jaxonnxruntime/onnx_ops/bitshift.py sha256=QGW1XRoUxM2YjsP_gIhbAliGX-9u5cy4v6hu71CGlJY 2071
jaxonnxruntime/onnx_ops/cast.py sha256=C8rpsPDjFQt9HHqh67pWrcTv04_w6tQPedDyONynYM8 4180
jaxonnxruntime/onnx_ops/castlike.py sha256=QFz42giZoLy5KEY5PwxDoZe0hX7dNGbM70i6LbupXhg 3057
jaxonnxruntime/onnx_ops/ceil.py sha256=qUZNnGEM9YtV1bIemX90IMUMOuxIWJICCJhY-EDsaJg 2091
jaxonnxruntime/onnx_ops/clip.py sha256=h_IEqAcBpG0F9aEny2BBDGCwDBkAcbO5i9QaCzoaJSM 2098
jaxonnxruntime/onnx_ops/concat.py sha256=725ptFbUI5MWDXbOSPU__64l_N9CSHx2JY7_Sf1VpXE 2809
jaxonnxruntime/onnx_ops/constant.py sha256=e3lv3tQXqAykcesWrHgI1cykwBbuq1bxJOqQ1gSUgk8 4368
jaxonnxruntime/onnx_ops/constantofshape.py sha256=YBOxvt1Wis1Z0xAPzlIoImjVsa5n-yYSCOWRhahdlfY 3060
jaxonnxruntime/onnx_ops/conv.py sha256=wsLYk0aP7YvQXJv4Xjd4Wx5Hx2fQT5cbxYizJNSbIMA 4555
jaxonnxruntime/onnx_ops/cos.py sha256=wic94a12wUbGrwenMCzWYEaFI0dz3Rxd3HhVjv5L9CY 1877
jaxonnxruntime/onnx_ops/cosh.py sha256=M1qXQNRIYpnLz0e5ieE53p9-Kifzi2l6LZJ0VjH7Elo 1887
jaxonnxruntime/onnx_ops/div.py sha256=B1dOSxHicU8mxJ2C4k0o0kkDZ9x_J34tUFrBew6Hk1g 3136
jaxonnxruntime/onnx_ops/dropout.py sha256=sb3zfgm6rh5SsVGGYEfqRFKiuM27djs0nxYPHybg2SA 3564
jaxonnxruntime/onnx_ops/einsum.py sha256=zUawPD7bfx1sQDntI3R7eCET1AmVgeOorBCSGu8L6oY 1894
jaxonnxruntime/onnx_ops/equal.py sha256=6eosDPOL7JiY10rgenxB15WBlumMI8br3_c3EWsr1lU 2933
jaxonnxruntime/onnx_ops/erf.py sha256=ljLduPDGzKvvIGedap6eBd3yO2y76RFArS0bGhUSCO4 1864
jaxonnxruntime/onnx_ops/exp.py sha256=hgbRS8K7KfK4h9cbfzIM7o0DOVqsvPJFI0vul4sPAY4 2811
jaxonnxruntime/onnx_ops/expand.py sha256=idLVo7StTVs2XlLg668E4WaohhM3cZa8Zy9PQklfqnk 3047
jaxonnxruntime/onnx_ops/flatten.py sha256=d9MUr2pw6EM3DPEQG0MADyuOsI5L9kXwfegiQroapQY 2631
jaxonnxruntime/onnx_ops/gather.py sha256=H2dkrt16U-fsLcwAx6C1ojFswZf5ter-sAeDDEKexGQ 2713
jaxonnxruntime/onnx_ops/gatherelements.py sha256=N3b3vChEtqf4m4tlCoyEsFH-UWs2xGcbVv0ZasCZAlA 1921
jaxonnxruntime/onnx_ops/gemm.py sha256=BIqpKV8nl7pqMDD-Ufnp3KVEErzRxssX5mfC4lfRZDg 3843
jaxonnxruntime/onnx_ops/globalaveragepool.py sha256=8Vmmz5U4e8nzrbx_oaMFyfiWLBm_LJi3fy-F3UhvDXs 2647
jaxonnxruntime/onnx_ops/identity.py sha256=VUfQeGTcybc8t6zbAq8PXGuL1gn4ULkotNHpnqxD5wM 3153
jaxonnxruntime/onnx_ops/if_op.py sha256=81U5pt4kfIB08Ie1FF2vG8ahmlIekmZzS8t9RRKFwZ8 6361
jaxonnxruntime/onnx_ops/leakyrelu.py sha256=uklwGtdMgWviFbwCQNVwa9ogVIjAV7gYEZvBQI1zS90 2839
jaxonnxruntime/onnx_ops/less.py sha256=gkTrnVPxPyFS6KdACuoXjeoQFeL4FAgp7PvY4Cv27II 2697
jaxonnxruntime/onnx_ops/lessorequal.py sha256=haGRLjviLZOPpX7NrjVOT_abiWu_J9K9yqrcQdr0gGk 2789
jaxonnxruntime/onnx_ops/log.py sha256=NR-VmiQIWZ5K2RYYh1q_ZRZYAvvz1mQs-oWY9RhpJOw 2093
jaxonnxruntime/onnx_ops/logsoftmax.py sha256=XypZ6PNNAs7VkJQwsnp_fw-uUaLx8lWc7eYVVv3R49w 2603
jaxonnxruntime/onnx_ops/lrn.py sha256=AaoeXPNDLV3MblIMoKzSp0PmY3EL2566mBQeSSHkgeg 3256
jaxonnxruntime/onnx_ops/matmul.py sha256=y9fbW1iQcfDlCnvDQsu9Lg5vhxMOanKSsDW3Zw4L4PQ 2862
jaxonnxruntime/onnx_ops/max.py sha256=CKDFyKfXRS-uCE_5XJ1eXlRv_aIH_FX51SOtu5Pa4qs 2816
jaxonnxruntime/onnx_ops/maxpool.py sha256=5dqUEUG2E9a1F_Hiu4Znn4vMLiUZu_-adA5jFVjzvNE 4764
jaxonnxruntime/onnx_ops/min.py sha256=Io8otVZnhVMJByyhpbNJ1y0LQrAaZLL6WmsVZp9lnMI 2225
jaxonnxruntime/onnx_ops/mul.py sha256=7FjS6rONEtDXCSx58qZNcH-1Hxv8FLwz4axWqvXDqkA 2443
jaxonnxruntime/onnx_ops/neg.py sha256=bnz-tV-lW39i_cMr57Dq5Ti2GaK_UuU2J_mGaZLKlVI 2083
jaxonnxruntime/onnx_ops/nonzero.py sha256=SWvqMlpPigYEHlvkHkCsxELD7NJ4bn_H0od-zB4dKEo 2478
jaxonnxruntime/onnx_ops/onehot.py sha256=bENdtD6RAFRJW4lWJIFc57_E6bEgeobfRG_XJaudyj8 2526
jaxonnxruntime/onnx_ops/onnx_not.py sha256=LlBdYGOv4YHpBMWnITwPSq2se11BWCVpiwEoXowKeB0 1362
jaxonnxruntime/onnx_ops/or_op.py sha256=-uDvoAXdihAg9eadRRsd3grQCGO04hB5QXSgUrwSV1Y 2467
jaxonnxruntime/onnx_ops/pad.py sha256=gevfLQyuP2YhrApJp7qN8WCYlgTiB3ZoCaLGf7X8ek4 4605
jaxonnxruntime/onnx_ops/pow.py sha256=f6FEMXDQxbq8VQYFF8fpV7P8tvM_yzYNJxZC6CrPU4c 2614
jaxonnxruntime/onnx_ops/prelu.py sha256=1wSc9aSZZWJKiIbQMyGNEvitAGN0yvCHQs27FUX8EBw 2147
jaxonnxruntime/onnx_ops/range.py sha256=adGTBx5TUbTVvAH_mFue8eW-0_s7iyH8dQR9fV0cS3U 3047
jaxonnxruntime/onnx_ops/reciprocal.py sha256=bO-eOdNpzTBax_87zUNuy6D3VX8mD0IzyS7ebHvgziI 2760
jaxonnxruntime/onnx_ops/reducemax.py sha256=RMUKxuVM5ctpf5CBpn7rd5ZFuxR8hpgaRD66qNgRMwM 3055
jaxonnxruntime/onnx_ops/reducemean.py sha256=iDE46p6H2Qi-N1TVfsbvYw8_vg9Z5FeLXpkbsfKCRL0 3537
jaxonnxruntime/onnx_ops/reducesum.py sha256=Tae_azH0LAC68pZXNEH4uw4rzcBjQOeXEVcFUlrz2HM 3093
jaxonnxruntime/onnx_ops/relu.py sha256=aBO6a6vNy2VBWS5EdIVFtlGtR6WlxpmfwmYu1XO06bE 2675
jaxonnxruntime/onnx_ops/reshape.py sha256=S7cZ07wyrFMTmQegNNxObkdXtqbNJ88QhchLN2TO4J8 3908
jaxonnxruntime/onnx_ops/selu.py sha256=s66BpYNyrzcxQFcmkubtYwhMXD0K1Fwq2-Z0HJR9kdU 1824
jaxonnxruntime/onnx_ops/shape.py sha256=jK0pkLPFKJURzujNKRMqac1q77e65Gwn8Ck8SFYCU_w 3135
jaxonnxruntime/onnx_ops/sigmoid.py sha256=QCGw0kYYVrvRatj7XHVzNkBsYIATiQZE5NjY-btKYqU 2706
jaxonnxruntime/onnx_ops/sin.py sha256=dcKBsafYjXYN7CpuErVGIN91QmvOyveeu8gZ6ClgDOk 1877
jaxonnxruntime/onnx_ops/sinh.py sha256=ZcXqI_V5ke4G_JuZh5yIQpoK5R5_9VDGj65Ml_ql1qQ 1887
jaxonnxruntime/onnx_ops/slice.py sha256=kB6qidE9pj-gxQAGA6x27I62uuWKYiXwsjN2rbwWWoQ 3626
jaxonnxruntime/onnx_ops/softmax.py sha256=fXRN7WNmNUhqs5eJz0fzStvPw3nAy1UhtVnaIszT0kg 2845
jaxonnxruntime/onnx_ops/softplus.py sha256=00AM1iLtNUuXg8XlC3T28fvN1R9Vtuoo5nSUq4d4To4 1912
jaxonnxruntime/onnx_ops/split.py sha256=SVYc0nQEV4DSlCPPgJHicbS7sHe9aIAk95jmkWO6tJI 3731
jaxonnxruntime/onnx_ops/sqrt.py sha256=9mF7SWf5MLkUiOiJJ3ErHxqX7oCWuioBeanW3ZjjpuI 2601
jaxonnxruntime/onnx_ops/squeeze.py sha256=7w_GjbdmmAfCAxOb8wRgzHaFrlNr_YPWyh-J7hNB1Lk 2714
jaxonnxruntime/onnx_ops/sub.py sha256=6NV9H7s77oPMjl8S1RpmjITaf2XL-EeHYblqnNkDvIM 3034
jaxonnxruntime/onnx_ops/sum.py sha256=WYRX5Q-AVhEOTtK-xNu5BwpI7e9eWxwSUKYRvjx209I 2868
jaxonnxruntime/onnx_ops/tanh.py sha256=Tg2IpyDBHK8hO6FvgEwARMWWueeh1snn8NO96xZWQZg 2611
jaxonnxruntime/onnx_ops/topk.py sha256=TQXT9iR2DxrwdTSgcL1lCS_N_1WiszLTPh2eN2GKLDo 3973
jaxonnxruntime/onnx_ops/transpose.py sha256=sgQIuTAraWBTRzyfcgaTRMTPScxKcxEW9YY0t_j7JAM 2692
jaxonnxruntime/onnx_ops/trilu.py sha256=AkWh2W_t_NAsMITZh2oK1tSHlXfcmjwmMuIAkbJiAY0 2453
jaxonnxruntime/onnx_ops/unsqueeze.py sha256=iM4-HrMHHFsa-zFq67OphF7d4nH1OVQEXxn8RxI4WF0 2787
jaxonnxruntime/onnx_ops/where.py sha256=bG1Nj9ZYYfm3ogLOLHMIjlMBRwPr8IEMg-Jm-gSLLhY 2718
jaxonnxruntime-0.3.0.dist-info/LICENSE sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0 11357
jaxonnxruntime-0.3.0.dist-info/METADATA sha256=hNXTnXessGQLo8QJKG1JYxXeHFnAwOna4oKsBv4659o 3076
jaxonnxruntime-0.3.0.dist-info/WHEEL sha256=5sUXSg9e4bi7lTLOHcm6QEYwO5TIF1TNbTSVFVjcJcc 92
jaxonnxruntime-0.3.0.dist-info/top_level.txt sha256=n-_9XNbgS37FAFgp7PIT9kw1o6Ud9-QHHXoLjoKH9CY 15
jaxonnxruntime-0.3.0.dist-info/RECORD

top_level.txt

jaxonnxruntime