JAX2ONNX

https://github.com/enpasos/jax2onnx

JAX2ONNX converts JAX, Flax NNX, and Equinox functions directly into the ONNX format, offering a...

Tokens:54,814
Snippets:651
Trust Score:5.6
License:Apache-2.0
Update:2 months ago
Tokens:
Raw