pipeline/train/requirements/train.in (2 lines of code) (raw):

opustrainer==0.4 gpustat==1.1.1