torch==2.6.0
triton==3.2.0
numpy
matplotlib
pandas
pytest
jupyter
ipykernel
IPython
