Adding (incomplete) documentation for MoE calculation.