numpy
networkx<=3.5,>=2.5
astunparse
sympy>=1.9
pyyaml
ply
fparser>=0.1.3
aenum>=3.1
dill
packaging

[:platform_system == "Windows"]
pyreadline

[:python_version < "3.7"]
dataclasses

[:python_version < "3.8"]
typing-compat

[docs]
jinja2<3.2.0
sphinx-autodoc-typehints
sphinx-rtd-theme>=0.5.1

[linting]
pre-commit==4.1.0
yapf==0.43.0

[ml]
onnx
torch
onnxsim
onnxscript
onnxruntime
protobuf
ninja

[ml-testing]
coverage
pytest-cov
scipy
absl-py
opt_einsum
pymlir
click
ipykernel
nbconvert
pytest-timeout
transformers==4.50
jax<=0.6.2
efficientnet_pytorch

[testing]
coverage
pytest-cov
scipy
absl-py
opt_einsum
pymlir
click
ipykernel
nbconvert
pytest-timeout
