Skip to content

0.14.1

Latest
Latest

Choose a tag to compare

View all tags
@enpasos enpasos released this 18 Jun 20:09
v0.14.1
25ba1f2

jax2onnx 0.14.1 — targeted Opset 27 coverage, centralized JAX compatibility, lower-bound CI validation, and refreshed runtime stack

  • Added targeted Opset 27 coverage:
    Kept Opset 23 as the default while enabling native FP16/BF16 Range exports for opset=27; older opset exports continue to use the cast-based fallback. Added coverage for new JAX 0.10 resize methods where the ONNX mapping is exact.

  • Centralized JAX internal API compatibility:
    Added a package-level JAX compatibility layer and migrated direct jax.core / jax.extend.core usage across converter and plugin code. Literal resolution is now guarded across JAX layout differences so future JAX internal API moves can be handled in one place.

  • Validated declared dependency lower bounds in CI:
    Raised the installable minimum JAX version to 0.8.1 for Flax/NNX 0.12.1 compatibility, added a minimum-dependency workflow job, and covered the lower-bound installer path with focused tests.

  • Refreshed the validation stack:
    Updated the documented runtime stack to JAX 0.10.2, Equinox 0.13.8, ONNX 1.22.0, and ONNX Runtime 1.27.0; Web validation remains on the latest stable onnxruntime-web release until a matching 1.27.x Web package is published.

Assets 2
Loading