Is there a good way this could be used for model distillation? Hmmm