[WIP] Adding training support#160

ParamThakkar123 wants to merge 1 commit intometa-pytorch:mainmeta-pytorch/BackendBench:mainfrom

ParamThakkar123:trainingParamThakkar123/BackendBench:trainingCopy head branch name to clipboard