Torch compile (16/21)

Torch compile