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.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.16.0.tar.gz | 782.6 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| jax2onnx-0.16.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 2.1 MB
Release files / jax2onnx-0.16.0.tar.gz
| Download URL | jax2onnx-0.16.0.tar.gz |
|---|---|
| Size | 782.6 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
45fcd4945318796ba8b84bfb0dcdca134207d0763469d588720c975193769a7f
|
|
BLAKE2b-256 checksum How to use checksums |
e1a858583ffc689d8e43a5fb9fb21d1e9926adad25cef910f7bc61c16783c004
|
| 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.0-py3-none-any.whl
| Download URL | jax2onnx-0.16.0-py3-none-any.whl |
|---|---|
| Size | 1.3 MB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
ff1ba97b2e6634333add35f972791bfd992f3587e900612d83c191110d0a71d5
|
|
BLAKE2b-256 checksum How to use checksums |
3a6abbed4bec60a9fe9213afea0c3852039d0890d625cfc66f2a5781112e39ea
|
| 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
|
