python/memory/measurments/act-mem-2/requirements.txt

einops==0.8.0
filelock==3.14.0
fsspec==2024.5.0
iniconfig==2.0.0
Jinja2==3.1.4
MarkupSafe==2.1.5
mpmath==1.3.0
networkx==3.3
packaging==24.0
pluggy==1.5.0
pytest==8.2.1
sympy==1.12.1
torch==2.3.0
typing_extensions==4.12.0