This is in Tensorflow. Would rather see a numpy version or something along those lines so that students can better understand what each step looks like in code.

I concur on the comments noting lack of explanation for the notation/lemmas/proof.