Torch compile (2/21)

Torch compile