lightning==2.0.7
mlflow==2.6.0
numpy==1.26.4
pydantic==1.10.12
setuptools>=68.0.0
torch>=2.0.1
torchmetrics>=1.0.0
torchvision>=0.15.2
onnx==1.14.1
torchsummary==1.5.1
modlee-onnx2torch==1.5.11
modlee-onnx_graphsurgeon==0.3.27
pymfe==0.4.2
pytest==8.1.1
nbsphinx
sphinx-rtd-theme
