Torch compile (20/21)

Torch compile