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.1
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.1.tar.gz | 796.4 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| jax2onnx-0.17.1-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 2.1 MB
Release files / jax2onnx-0.17.1.tar.gz
| Download URL | jax2onnx-0.17.1.tar.gz |
|---|---|
| Size | 796.4 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
c86175e63df07580827d9c015d5dc6dc484e58027abee3c34bf786644934fc67
|
|
BLAKE2b-256 checksum How to use checksums |
bce0084d4962bd34705dbf5764c100f221294c1e6bd50685160c07746924dbcc
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
poetry/2.1.3 CPython/3.12.5 Linux/6.18.40.1-microsoft-standard-WSL2
|
Release files / jax2onnx-0.17.1-py3-none-any.whl
| Download URL | jax2onnx-0.17.1-py3-none-any.whl |
|---|---|
| Size | 1.4 MB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
ca4ad1adc58bb9eed8ffd13064fafeec9b81cbc7eeca86ad913f37fcd67b2c49
|
|
BLAKE2b-256 checksum How to use checksums |
372e2105a6cfa77b2a130c559fb09ceb7b2045b29d22372c7764d30e907b07ec
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
poetry/2.1.3 CPython/3.12.5 Linux/6.18.40.1-microsoft-standard-WSL2
|
