packaging
wheel
psutil
numpy
# torch version needs to pinned here for CUDA builds
torch==2.11.0
apache-tvm-ffi==0.1.13
tokenspeed-proton>=3.8.10.post20260721
tokenspeed-triton>=3.8.10.post20260721
# The 4.6 libs wheels declare protobuf<7 solely for the (unused) iket
# profiler gencode; serving needs protobuf>=7.35.1 for the smg grpc-health
# gencode. Keep protobuf 7.x and ignore that resolver warning.
nvidia-cutlass-dsl[cu13]==4.6.0
nvidia-cutlass-dsl-libs-cu13==4.6.0
flashinfer-python==0.6.16
nvidia-ml-py==13.580.82
nvtx
