This is the version of the Training Distribution Matching code repository that corresponds to the submitted preprint and manuscript.