Skip to content

Dependencies

Latest supported versions of major dependencies:

Library Version
JAX 0.9.2
Flax 0.12.6
Equinox 0.13.6
onnx-ir 0.2.0
onnx 1.21.0
onnxruntime 1.24.4

For minimum supported versions and optional extras, see pyproject.toml. For the fully resolved Poetry environment, see poetry.lock.