Seriously, I've been waiting for a long time now for this to come about. This would make pytorch a much more powerful platform.