Compiles the model with Torch. Experimental!