# pytorch
--index-url https://download.pytorch.org/whl/rocm6.2
torch==2.5.1+rocm6.2
torchvision==0.20.1+rocm6.2
onnxruntime-gpu==1.19.2

# xformers
# TODO

# optimizers
# TODO
