piwheels

jax2onnx

export JAX to ONNX

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install jax2onnx

Dependencies

• huggingface-hub
• netron
• dm-pix
• flax
• jax
• onnx
• onnx-ir
• einops
• orbax-checkpoint
• onnxruntime
• equinox
• ml-dtypes
• optax
• orbax-export

Releases

Version	Released	Bullseye Python 3.9	Bookworm Python 3.11	Trixie Python 3.13	Files
0.13.1	2026-05-05
jax2onnx-0.13.1-py3-none-any.whl (1 MB)
0.13.0	2026-04-22
jax2onnx-0.13.0-py3-none-any.whl (1 MB)
0.12.5	2026-04-02
jax2onnx-0.12.5-py3-none-any.whl (1 MB)
0.12.4	2026-03-28
jax2onnx-0.12.4-py3-none-any.whl (1 MB)
0.12.3	2026-03-04
jax2onnx-0.12.3-py3-none-any.whl (1 MB)
0.12.2	2026-02-28
jax2onnx-0.12.2-py3-none-any.whl (1 MB)
0.12.1	2026-02-21
jax2onnx-0.12.1-py3-none-any.whl (884 KB)
0.12.0	2026-02-09
jax2onnx-0.12.0-py3-none-any.whl (777 KB)
0.11.2	2026-01-26
jax2onnx-0.11.2-py3-none-any.whl (749 KB)
0.11.1	2026-01-13
jax2onnx-0.11.1-py3-none-any.whl (755 KB)
0.11.0	2025-12-29
jax2onnx-0.11.0-py3-none-any.whl (736 KB)
0.10.4	2025-12-22
jax2onnx-0.10.4-py3-none-any.whl (660 KB)
0.10.3	2025-12-09
jax2onnx-0.10.3-py3-none-any.whl (653 KB)
0.10.2	2025-12-01
jax2onnx-0.10.2-py3-none-any.whl (646 KB)
0.10.1	2025-11-06
jax2onnx-0.10.1-py3-none-any.whl (607 KB)
0.10.0	2025-10-24
jax2onnx-0.10.0-py3-none-any.whl (565 KB)
0.9.0	2025-10-12
jax2onnx-0.9.0-py3-none-any.whl (484 KB)
0.8.1	2025-09-06
jax2onnx-0.8.1-py3-none-any.whl (541 KB)
0.8.0	2025-08-25
jax2onnx-0.8.0-py3-none-any.whl (530 KB)
0.7.5	2025-08-20
jax2onnx-0.7.5-py3-none-any.whl (513 KB)
0.7.4	2025-08-03
jax2onnx-0.7.4-py3-none-any.whl (459 KB)
0.7.3	2025-07-31
jax2onnx-0.7.3-py3-none-any.whl (454 KB)
0.7.2	2025-07-28
jax2onnx-0.7.2-py3-none-any.whl (452 KB)
0.7.1	2025-07-12
jax2onnx-0.7.1-py3-none-any.whl (440 KB)
0.7.0	2025-07-01
jax2onnx-0.7.0-py3-none-any.whl (380 KB)
0.6.5	2025-06-17
jax2onnx-0.6.5-py3-none-any.whl (370 KB)
0.6.4	2025-06-11
jax2onnx-0.6.4-py3-none-any.whl (355 KB)
0.6.3	2025-06-08
jax2onnx-0.6.3-py3-none-any.whl (354 KB)
0.6.2	2025-06-07
jax2onnx-0.6.2-py3-none-any.whl (350 KB)
0.6.1	2025-06-06
jax2onnx-0.6.1-py3-none-any.whl (348 KB)
0.6.0	2025-05-24
jax2onnx-0.6.0-py3-none-any.whl (321 KB)
0.5.2	2025-05-12
jax2onnx-0.5.2-py3-none-any.whl (292 KB)
0.5.1	2025-05-08
jax2onnx-0.5.1-py3-none-any.whl (280 KB)
0.5.0	2025-05-06
jax2onnx-0.5.0-py3-none-any.whl (268 KB)
0.4.4	2025-04-30
jax2onnx-0.4.4-py3-none-any.whl (216 KB)
0.4.3	2025-04-21
jax2onnx-0.4.3-py3-none-any.whl (209 KB)
0.4.2	2025-04-16
jax2onnx-0.4.2-py3-none-any.whl (197 KB)
0.4.1	2025-04-11
jax2onnx-0.4.1-py3-none-any.whl (182 KB)
0.4.0	2025-04-11
jax2onnx-0.4.0-py3-none-any.whl (182 KB)
0.3.2	2025-04-08
jax2onnx-0.3.2-py3-none-any.whl (114 KB)
0.3.1	2025-04-08
jax2onnx-0.3.1-py3-none-any.whl (106 KB)
0.3.0	2025-03-20
jax2onnx-0.3.0-py3-none-any.whl (114 KB)
0.2.0	2025-03-17
jax2onnx-0.2.0-py3-none-any.whl (107 KB)

Issues with this package?

• Search issues for this package
• Package or version missing? Open a new issue
• Something else? Open a new issue