Torch compile (3/21)

Torch compile