code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation" - View it on GitHub