numpy
jax
jaxlib

[all]
onnx==1.12.0

[testing]
mypy
pytest
pytest-cov
pytest-custom_exit_code
pytest-xdist
absl-py
