onnx
onnxruntime==1.20.0
pydantic
matplotlib
numpy
pathlib

[all]
black
darglint
mypy
pytest
ruff
types-tensorflow
torch
tensorflow
tf2onnx>=1.16.0
jax
equinox
numpy<2

[dev]
black
darglint
mypy
pytest
ruff
types-tensorflow

[jax]
tensorflow
tf2onnx>=1.16.0
jax
equinox
numpy<2

[pytorch]
torch
