fbgemm-gpu
tensordict
torchmetrics==1.0.3
tqdm
pyre-extensions
iopath
