Add torch framework definition