onnx
onnxruntime==1.20.0

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

[dev]
black
darglint
mypy
pytest
ruff
types-tensorflow

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

[pytorch]
torch

[vis]
matplotlib
numpy
