jax2onnx 🌟
jax2onnx converts your JAX, Flax NNX, Flax Linen, Equinox functions directly into the ONNX format.
📚 Documentation
Read the full documentation here
🚀 Quick Install
pip install jax2onnx
⚡ Quick Usage
from jax2onnx import to_onnx
from flax import nnx
model = MyFlaxModel(...)
to_onnx(model, [("B", 32)], return_mode="file", output_path="model.onnx")
🤝 Contributing
We warmly welcome contributions! Please check our Developer Guide for plugin tutorials and architecture details.
📜 License
Apache License, Version 2.0. See LICENSE.
🌟 Special Thanks
A huge thank you to all our contributors and the community for their help and inspiration!
Happy converting! 🎉
1 maintainer
Metadata
1 maintainer
Release files for jax2onnx 0.17.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| jax2onnx-0.17.0.tar.gz | 793.3 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| jax2onnx-0.17.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 2.1 MB
Release files / jax2onnx-0.17.0.tar.gz
| Download URL | jax2onnx-0.17.0.tar.gz |
|---|---|
| Size | 793.3 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
669209dce300fed70439ae93493b024312121244fdadc3fb81b536d9e3a758be
|
|
BLAKE2b-256 checksum How to use checksums |
6752bc28aa8a3631d3a03f253f21ec252671f69c5663b40e0bf82bc9086b3c43
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
poetry/2.1.3 CPython/3.12.5 Linux/6.18.33.2-microsoft-standard-WSL2
|
Release files / jax2onnx-0.17.0-py3-none-any.whl
| Download URL | jax2onnx-0.17.0-py3-none-any.whl |
|---|---|
| Size | 1.3 MB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
aeee6a8ef32de37b3d9335ba264e8e99b1e7518bc8b1e3037d20fbd3b46726dd
|
|
BLAKE2b-256 checksum How to use checksums |
7c5f8a60e3abe1c4797a21f136fd76cd219e42aa353b847df04088fe78c45bd8
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
poetry/2.1.3 CPython/3.12.5 Linux/6.18.33.2-microsoft-standard-WSL2
|
