cupy-cuda12x
triton==3.2.0; (platform_system != "Windows")
torch
--extra-index-url https://pypi.ngc.nvidia.com
polygraphy
colored
numpy==1.26.4
pyyaml==6.0.1
requests==2.32.2
tqdm==4.66.4
