Code to reproduce results from the CellRank preprint - View it on GitHub