rocm.txt

# Common dependencies
-r common.txt

numba == 0.60.0; python_version == '3.9' # v0.61 doesn't support Python 3.9. Required for N-gram speculative decoding
numba == 0.61.2; python_version > '3.9'

# Dependencies for hcus
boto3
botocore
datasets
ray[cgraph]>=2.48.0 # Ray Compiled Graph, required for pipeline parallelism in V1. 
peft
pytest-asyncio
tensorizer==2.10.1
packaging>=24.2
setuptools>=77.0.3,<80.0.0
setuptools-scm>=8
runai-model-streamer==0.11.0
runai-model-streamer-s3==0.11.0
# conch-triton-kernels==1.2.1 # numpy>=1.26.4
timm>=1.0.17

numa
pytrie
setuptools_scm>=8
cmake==3.29
quart
fastrlock==0.8.3
cupy==12.3.0

torch>=2.5.1,<=2.7.1
torchvision>=0.20.1,<=0.22.0
triton == 3.1
flash_attn == 2.6.1
flash_mla == 1.0.0
lightop == 0.6.0
lmslim == 0.3.1