pytorch-lightning==2.4.0
lightning==2.3.3
mlflow==2.16.2
networkx==2.6.3
numpy==1.26.4
pydantic==2.8.2
setuptools>=68.0.0
torch==2.3.0
torchmetrics==1.4.2
torchvision==0.18.0
torchtext==0.18.0
torchaudio==2.3.0
portalocker==2.10.1
torchdata==0.8.0
spacy==3.7.6
onnx==1.16.2
torchsummary==1.5.1
modlee-onnx2torch==1.5.15
modlee-onnx_graphsurgeon==0.3.27
pymfe==0.4.3
scipy==1.12.0
pydot==2.0.0
transformers==4.41.0
wonderwords

[docs]
nbsphinx
sphinx-rtd-theme

[testing]
pytest==8.1.1
