The code for the paper can be found here:
https://github.com/christopher-beckham/msc/tree/master/experiments/earth_mover
The Poisson and binomial extension functions/layers can be found:
For example, the binomial one is:
I haven't had time to do a nice clean-up, but let me know if you have any questions.