let hf trainer handle torch compile (#516)

let hf trainer handle torch compile (#516) #1670