Please see my Github page for more general Python versions of the code. The repository also has an implementation of two estimation procedures explained in our paper. They estimates the parameters of a semilinear surplus and the expected utilities for the Choo and Siow 2006 model (homoskedastic, with singles) and more generally for models with a given generalized entropy.
My coauthor Alfred Galichon (NYU) and Keith O’Hara (Amazon) have developed the TraME library: a set of procedures for Transportation Methods for Econometrics. It allows the user to estimate a much broader class of models. TraME is available on Github. Here are some more specific links: