"""Distillation module for knowledge distillation in Keras."""