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
Release files for jax2onnx 0.16.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.16.1.tar.gz | 783.1 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| jax2onnx-0.16.1-py3-none-any.whl | Python 3 | none | any | Details |
Total release size:2.1 MB
Release files / jax2onnx-0.16.1.tar.gz
| Download URL | jax2onnx-0.16.1.tar.gz |
|---|---|
| Size | 783.1 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
eee94a6b5387243b5e67865e959f793edabb9b2126e9dafdb898d9edd7ff442e
|
|
BLAKE2b-256 checksum How to use checksums |
8f64c8fb6919c91b6d7ec81c6469360da7a417ea799322aafc443342ba82224e
|
| 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.16.1-py3-none-any.whl
| Download URL | jax2onnx-0.16.1-py3-none-any.whl |
|---|---|
| Size | 1.3 MB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
899280eae53989dede7aa439d0a12e77fdc7baa95ea2417c9532915925eba181
|
|
BLAKE2b-256 checksum How to use checksums |
4d9736e0dd07742d2bceeac32833ed0d5d744c9ca4fa1ea2dab1759cf098adb9
|
| 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
|
