regex
torch~=2.2.1
numpy
matplotlib
# Includes a bugfix for recent NCCL versions
HolisticTraceAnalysis @ git+https://github.com/nelson-liu/HolisticTraceAnalysis.git@comm_kernel_fix