Currently JTP substitution works but gets error for GPU with mx vm