Implement KL Divergence | TensorTonic