So that's how model distillation is done. Just ask for the source code directly.