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.15.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.15.1.tar.gz | 777.1 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| jax2onnx-0.15.1-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 2.1 MB
Release files / jax2onnx-0.15.1.tar.gz
| Download URL | jax2onnx-0.15.1.tar.gz |
|---|---|
| Size | 777.1 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
6cf29df65f0445a3f0fda4b578c8459a8b690f18f1934d3dc4238f73fd3de5ee
|
|
BLAKE2b-256 checksum How to use checksums |
26b0e7a8eb40d901973905c30399732f39dbbc83aebbafe809eb1b580fd6eed1
|
| 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.15.1-py3-none-any.whl
| Download URL | jax2onnx-0.15.1-py3-none-any.whl |
|---|---|
| Size | 1.3 MB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
dd4fcdf39925fc2003115a32b2e1d00cb2da5dc17490e986f4eda1c3cf30d38d
|
|
BLAKE2b-256 checksum How to use checksums |
cb897e86ee0c4aed684f8fd8d2f5a45fab8ee04b8d8b8233089330c52a88540a
|
| 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
|
