@article{chen2023deep,
  title={Deep momentum multi-marginal Schr{\"o}dinger bridge},
  author={Chen, Tianrong and Liu, Guan-Horng and Tao, Molei and Theodorou, Evangelos},
  journal={Advances in Neural Information Processing Systems},
  volume={36},
  year={2024}
}

@book{kloeden1992stochastic,
  title={Stochastic differential equations},
  author={Kloeden, Peter E and Platen, Eckhard and Kloeden, Peter E and Platen, Eckhard},
  year={1992},
  publisher={Springer}
}

@article{mutti2023challenging,
  title={Challenging common assumptions in convex reinforcement learning},
  author={Mutti, Mirco and De Santi, Riccardo and De Bartolomeis, Piersilvio and Restelli, Marcello},
  journal={Advances in Neural Information Processing Systems},
  volume={35},
  pages={4489--4502},
  year={2022}
}

@article{schulman2017proximal,
      title={Proximal Policy Optimization Algorithms}, 
      author={John Schulman and Filip Wolski and Prafulla Dhariwal and Alec Radford and Oleg Klimov},
      year={2017},
journal={arXiv preprint arXiv:1707.06347},
}

@article{terpin2022trust,
  title={Trust Region Policy Optimization with Optimal Transport Discrepancies: Duality and Algorithm for Continuous Actions},
  author={Terpin, Antonio and Lanzetti, Nicolas and Yardim, Batuhan and Dorfler, Florian and Ramponi, Giorgia},
  journal={Advances in Neural Information Processing Systems},
  volume={35},
  pages={19786--19797},
  year={2022}
}

@article{zaheer2018deep,
  title={Deep sets},
  author={Zaheer, Manzil and Kottur, Satwik and Ravanbakhsh, Siamak and Poczos, Barnabas and Salakhutdinov, Russ R and Smola, Alexander J},
  journal={Advances in Neural Information Processing Systems},
  volume={30},
  year={2017}
}

@inproceedings{lee2019set,
  title={Set transformer: A framework for attention-based permutation-invariant neural networks},
  author={Lee, Juho and Lee, Yoonho and Kim, Jungtaek and Kosiorek, Adam and Choi, Seungjin and Teh, Yee Whye},
  booktitle={International conference on machine learning},
  pages={3744--3753},
  year={2019},
  organization={PMLR}
}

@article{vaswani2023attention,
  title={Attention is all you need},
  author={Vaswani, Ashish and Shazeer, Noam and Parmar, Niki and Uszkoreit, Jakob and Jones, Llion and Gomez, Aidan N and Kaiser, {\L}ukasz and Polosukhin, Illia},
  journal={Advances in Neural Information Processing Systems},
  volume={30},
  year={2017}
}

@article{flamary2021pot,
  author  = {R{\'e}mi Flamary and Nicolas Courty and Alexandre Gramfort and Mokhtar Z. Alaya and Aur{\'e}lie Boisbunon and Stanislas Chambon and Laetitia Chapel and Adrien Corenflos and Kilian Fatras and Nemo Fournier and L{\'e}o Gautheron and Nathalie T.H. Gayraud and Hicham Janati and Alain Rakotomamonjy and Ievgen Redko and Antoine Rolet and Antony Schutz and Vivien Seguy and Danica J. Sutherland and Romain Tavenard and Alexander Tong and Titouan Vayer},
  title   = {POT: Python Optimal Transport},
  journal = {Journal of Machine Learning Research},
  year    = {2021},
  volume  = {22},
  number  = {78},
  pages   = {1-8}
}



@article{cuturi2022optimal,
      title={Optimal Transport Tools ({OTT}): A {JAX} Toolbox for all things {W}asserstein}, 
      author={Marco Cuturi and Laetitia Meng-Papaxanthos and Yingtao Tian and Charlotte Bunne and Geoff Davis and Olivier Teboul},
      year={2022},
journal={arXiv preprint arXiv:2201.12324},
}

@book{bishop2006pattern,
  title={Pattern recognition and machine learning},
  author={Bishop, Christopher M and Nasrabadi, Nasser M},
  volume={4},
  number={4},
  year={2006},
  publisher={Springer}
}

@misc{scipy2020,
  author  = {Virtanen, Pauli and Gommers, Ralf and Oliphant, Travis E. and
            Haberland, Matt and Reddy, Tyler and Cournapeau, David and
            Burovski, Evgeni and Peterson, Pearu and Weckesser, Warren and
            Bright, Jonathan and {van der Walt}, St{\'e}fan J. and
            Brett, Matthew and Wilson, Joshua and Millman, K. Jarrod and
            Mayorov, Nikolay and Nelson, Andrew R. J. and Jones, Eric and
            Kern, Robert and Larson, Eric and Carey, C J and
            Polat, {\.I}lhan and Feng, Yu and Moore, Eric W. and
            {VanderPlas}, Jake and Laxalde, Denis and Perktold, Josef and
            Cimrman, Robert and Henriksen, Ian and Quintero, E. A. and
            Harris, Charles R. and Archibald, Anne M. and
            Ribeiro, Ant{\^o}nio H. and Pedregosa, Fabian and
            {van Mulbregt}, Paul and {SciPy 1.0 Contributors}},
  title   = {{{SciPy} 1.0: Fundamental Algorithms for Scientific
            Computing in Python}},
  journal = {Nature Methods},
  year    = {2020},
  volume  = {17},
  pages   = {261--272},
}

@inproceedings{paszke2017automatic,
  title={Automatic differentiation in PyTorch},
  author={Paszke, Adam and Gross, Sam and Chintala, Soumith and Chanan, Gregory and Yang, Edward and DeVito, Zachary and Lin, Zeming and Desmaison, Alban and Antiga, Luca and Lerer, Adam},
  booktitle={NIPS-W},
  year={2017}
}

@misc{jax2018github,
  author = {James Bradbury and Roy Frostig and Peter Hawkins and Matthew James Johnson and Chris Leary and Dougal Maclaurin and George Necula and Adam Paszke and Jake Vander{P}las and Skye Wanderman-{M}ilne and Qiao Zhang},
  title = {{JAX}: composable transformations of {P}ython+{N}um{P}y programs},
  year = {2018},
}

@inproceedings{adam2015,
  author       = {Diederik P. Kingma and
                  Jimmy Ba},
  editor       = {Yoshua Bengio and
                  Yann LeCun},
  title        = {Adam: {A} Method for Stochastic Optimization},
  booktitle    = {3rd International Conference on Learning Representations, {ICLR} 2015,
                  San Diego, CA, USA, May 7-9, 2015, Conference Track Proceedings},
  year         = {2015},
}

@article{alvarez2021optimizing,
title={Optimizing Functionals on the Space of Probabilities with Input Convex Neural Networks},
author={David Alvarez-Melis and Yair Schiff and Youssef Mroueh},
journal={Transactions on Machine Learning Research},
year={2022}
}

@article{peyré2020computational,
  title={Computational optimal transport: With applications to data science},
  author={Peyr{\'e}, Gabriel and Cuturi, Marco and others},
  journal={Foundations and Trends in Machine Learning},
  volume={11},
  number={5-6},
  pages={355--607},
  year={2019},
  publisher={Now Publishers, Inc.}
}

@article{brenier1991factorization,
author = {Brenier, Yann},
title = {Polar factorization and monotone rearrangement of vector-valued functions},
journal = {Communications on Pure and Applied Mathematics},
volume = {44},
number = {4},
pages = {375-417},
year = {1991}
}


@article{dephilippis2012second,
      title={Second order stability for the Monge-Ampere equation and strong Sobolev convergence of optimal transport maps}, 
      author={Guido De Philippis and Alessio Figalli},
      year={2012},
journal={arXiv preprint arXiv:1202.5561},
}

@article{terpin2023dynamic,
  title={Dynamic programming in probability spaces via optimal transport},
  author={Terpin, Antonio and Lanzetti, Nicolas and D{\"o}rfler, Florian},
  journal={SIAM Journal on Control and Optimization},
  volume={62},
  number={2},
  pages={1183--1206},
  year={2024},
  publisher={SIAM}
}

@inproceedings{weeratunga2003comparison,
  title={Comparison of PDE-based nonlinear anisotropic diffusion techniques for image denoising},
  author={Weeratunga, Sisira K and Kamath, Chandrika},
  booktitle={Image Processing: Algorithms and Systems II},
  volume={5014},
  pages={201--212},
  year={2003},
  organization={SPIE}
}

@inproceedings{li2020scalable,
  title={Scalable gradients for stochastic differential equations},
  author={Li, Xuechen and Wong, Ting-Kam Leonard and Chen, Ricky TQ and Duvenaud, David},
  booktitle={International Conference on Artificial Intelligence and Statistics},
  pages={3870--3882},
  year={2020},
  organization={PMLR}
}

@book{reece2017campbell,
  title={Campbell biology},
  author={Reece, Jane and Urry, Lisa and Cain, Michael},
  year={2017},
  publisher={Pearson}
}

@article{ho2020denoising,
  title={Denoising diffusion probabilistic models},
  author={Ho, Jonathan and Jain, Ajay and Abbeel, Pieter},
  journal={Advances in Neural Information Processing Systems},
  volume={33},
  pages={6840--6851},
  year={2020}
}

@article{geshkovski2023mathematical,
      title={A mathematical perspective on Transformers}, 
      author={Borjan Geshkovski and Cyril Letrouit and Yury Polyanskiy and Philippe Rigollet},
      year={2023},
journal={arXiv preprint arXiv:2312.10794},
}

@article{lanzetti2024,
      title={Variational Analysis in the {W}asserstein Space}, 
      author={Nicolas Lanzetti and Antonio Terpin and Florian Dörfler},
      year={2024},
    journal = {arXiv preprint arXiv:2406.10676}, 
}

@article{jordan1998variational,
  title={The variational formulation of the {F}okker--{P}lanck equation},
  author={Jordan, Richard and Kinderlehrer, David and Otto, Felix},
  journal={SIAM journal on mathematical analysis},
  volume={29},
  number={1},
  pages={1--17},
  year={1998},
  publisher={SIAM}
}

@misc{sfu_optimization,
  author = {{Simon Fraser University}},
  title = {Test Functions and Datasets for Optimization},
  year = {2023},
  url = {https://www.sfu.ca/~ssurjano/optimization.html},
  note = {Accessed: 2024-05-15}
}


@article{gazdieva2022optimal,
      title={An Optimal Transport Perspective on Unpaired Image Super-Resolution}, 
      author={Milena Gazdieva and Litu Rout and Alexander Korotin and Andrey Kravchenko and Alexander Filippov and Evgeny Burnaev},
      year={2022},
journal = {arXiv preprint arXiv:2202.01116}, 
}

@article{amos2023amortizing,
      title={On amortizing convex conjugates for optimal transport}, 
      author={Brandon Amos},
      year={2023},
journal = {arXiv preprint arXiv:2210.12153}, 
}

@article{schiebinger2019optimal,
  title={Optimal-transport analysis of single-cell gene expression identifies developmental trajectories in reprogramming},
  author={Schiebinger, Geoffrey and Shu, Jian and Tabaka, Marcin and Cleary, Brian and Subramanian, Vidya and Solomon, Aryeh and Gould, Joshua and Liu, Siyan and Lin, Stacie and Berube, Peter and others},
  journal={Cell},
  volume={176},
  number={4},
  pages={928--943},
  year={2019},
  publisher={Elsevier}
}

@article{yang2020predicting,
  title={Predicting cell lineages using autoencoders and optimal transport},
  author={Yang, Karren Dai and Damodaran, Karthik and Venkatachalapathy, Saradha and Soylemezoglu, Ali C and Shivashankar, GV and Uhler, Caroline},
  journal={PLoS computational biology},
  volume={16},
  number={4},
  pages={e1007828},
  year={2020},
  publisher={Public Library of Science San Francisco, CA USA}
}

@article{bonalli2023non,
  title={Non-Parametric Learning of Stochastic Differential Equations with Fast Rates of Convergence},
  author={Bonalli, Riccardo and Rudi, Alessandro},
  journal={arXiv preprint arXiv:2305.15557},
  year={2023}
}

@book{villani2009optimal,
  title={Optimal transport: old and new},
  author={Villani, C{\'e}dric},
  volume={338},
  year={2009},
  publisher={Springer}
}

@article{tong2023improving,
  title={Improving and generalizing flow-based generative models with minibatch optimal transport},
  author={Tong, Alexander and FATRAS, Kilian and Malkin, Nikolay and Huguet, Guillaume and Zhang, Yanlei and Rector-Brooks, Jarrid and Wolf, Guy and Bengio, Yoshua},
year={2024},
  journal={Transactions on Machine Learning Research}
}

@inproceedings{tong2020trajectorynet,
title={Trajectorynet: A dynamic optimal transport network for modeling cellular dynamics},
author={Tong, Alexander and Huang, Jessie and Wolf, Guy and Van Dijk, David and Krishnaswamy, Smita},
booktitle={International Conference on Machine Learning},
pages={9526--9536},
year={2020},
organization={PMLR}
}


@inproceedings{finlay2020train,
title={How to train your neural ODE: the world of Jacobian and kinetic regularization},
author={Finlay, Chris and Jacobsen, J{\"o}rn-Henrik and Nurbekyan, Levon and Oberman, Adam},
booktitle={International Conference on Machine Learning},
pages={3154--3164},
year={2020},
organization={PMLR}
}
@article{de2021diffusion,
title={Diffusion {S}chr{\"o}dinger bridge with applications to score-based generative modeling},
author={De Bortoli, Valentin and Thornton, James and Heng, Jeremy and Doucet, Arnaud},
journal={Advances in Neural Information Processing Systems},
volume={34},
pages={17695--17709},
year={2021}
}

 @article{koshizuka2022neural,
title={Neural {L}agrangian {Schrödinger} Bridge: Diffusion Modeling for Population Dynamics},
author={Koshizuka, Takeshi and Sato, Issei},
journal={arXiv preprint arXiv:2204.04853},
year={2022}
}

@inproceedings{bunne2022proximal,
  title={Proximal Optimal Transport Modeling of Population Dynamics},
  author={Bunne, Charlotte and Meng-Papaxanthos, Laetitia and Krause, Andreas and Cuturi, Marco},
  booktitle={International Conference on Artificial Intelligence and Statistics (AISTATS)},
  year={2022},
}

@article{bunne2023single-cell-perturbation,
  title={Learning single-cell perturbation responses using neural optimal transport},
  author={Bunne, Charlotte and Stark, Stefan G and Gut, Gabriele and Del Castillo, Jacobo Sarabia and Levesque, Mitch and Lehmann, Kjong-Van and Pelkmans, Lucas and Krause, Andreas and R{\"a}tsch, Gunnar},
  journal={Nature Methods},
  pages={1--10},
  year={2023},
  publisher={Nature Publishing Group US New York}
}

@article{huguet2022manifold,
  title={Manifold interpolating optimal-transport flows for trajectory inference},
  author={Huguet, Guillaume and Magruder, Daniel Sumner and Tong, Alexander and Fasina, Oluwadamilola and Kuchroo, Manik and Wolf, Guy and Krishnaswamy, Smita},
  journal={Advances in Neural Information Processing Systems},
  volume={35},
  pages={29705--29718},
  year={2022}
}

@inproceedings{scarvelis2022riemannian,
  title={Riemannian Metric Learning via Optimal Transport},
  author={Scarvelis, Christopher and Solomon, Justin},
    year={2023},
  booktitle={The Eleventh International Conference on Learning Representations}
}


@article{jko1998,
author = {Jordan, Richard and Kinderlehrer, David and Otto, Felix},
title = {The Variational Formulation of the Fokker--Planck Equation},
journal = {SIAM Journal on Mathematical Analysis},
volume = {29},
number = {1},
pages = {1-17},
year = {1998}
}

@article{brenier1991polar,
  title={Polar factorization and monotone rearrangement of vector-valued functions},
  author={Brenier, Yann},
  journal={Communications on pure and applied mathematics},
  volume={44},
  number={4},
  pages={375--417},
  year={1991},
  publisher={Wiley Online Library}
}

@article{alphafold2021,
  title={Highly accurate protein structure prediction with AlphaFold},
  author={Jumper, John and Evans, Richard and Pritzel, Alexander and Green, Tim and Figurnov, Michael and Ronneberger, Olaf and Tunyasuvunakool, Kathryn and Bates, Russ and {\v{Z}}{\'\i}dek, Augustin and Potapenko, Anna and others},
  journal={Nature},
  volume={596},
  number={7873},
  pages={583--589},
  year={2021},
  publisher={Nature Publishing Group}
}

@article{biomedDeliverySurgerySensingDetox,
author = {Jinxing Li  and Berta Esteban-Fernández de Ávila  and Wei Gao  and Liangfang Zhang  and Joseph Wang },
title = {Micro/nanorobots for biomedicine: Delivery, surgery, sensing, and detoxification},
journal = {Science Robotics},
volume = {2},
number = {4},
pages = {eaam6431},
year = {2017},
}


@article{rocketScienceNanoscale,
author = {Li, Jinxing and Rozen, Isaac and Wang, Joseph},
title = {Rocket Science at the Nanoscale},
journal = {ACS Nano},
volume = {10},
number = {6},
pages = {5619-5634},
year = {2016},
}


@book{ambrosio2005gradient,
  title={Gradient flows: in metric spaces and in the space of probability measures},
  author={Ambrosio, Luigi and Gigli, Nicola and Savar{\'e}, Giuseppe},
  year={2005},
  publisher={Springer Science \& Business Media}
}


@book{figalli2021invitation,
  title={An invitation to optimal transport, {W}asserstein distances, and gradient flows},
  author={Figalli, Alessio and Glaudo, Federico},
  year={2021}
}


@article{peyre2019computational,
  title={Computational optimal transport: With applications to data science},
  author={Peyr{\'e}, Gabriel and Cuturi, Marco and others},
  journal={Foundations and Trends{\textregistered} in Machine Learning},
  volume={11},
  number={5-6},
  pages={355--607},
  year={2019},
  publisher={Now Publishers, Inc.}
}

@inproceedings{uscidda2023monge,
  title={The {M}onge gap: A regularizer to learn all transport maps},
  author={Uscidda, Th{\'e}o and Cuturi, Marco},
  booktitle={International Conference on Machine Learning},
  pages={34709--34733},
  year={2023},
  organization={PMLR}
}


@article{bunne2022supervised,
  title={Supervised training of conditional {M}onge maps},
  author={Bunne, Charlotte and Krause, Andreas and Cuturi, Marco},
  journal={Advances in Neural Information Processing Systems},
  volume={35},
  pages={6859--6872},
  year={2022}
}


@article{grathwohl2018ffjord,
  title={Ffjord: Free-form continuous dynamics for scalable reversible generative models},
  author={Grathwohl, Will and Chen, Ricky TQ and Bettencourt, Jesse and Sutskever, Ilya and Duvenaud, David},
  journal={arXiv preprint arXiv:1810.01367},
  year={2018}
}


@inproceedings{hashimoto2016learning,
  title={Learning population-level diffusions with generative {RNNs}},
  author={Hashimoto, Tatsunori and Gifford, David and Jaakkola, Tommi},
  booktitle={International Conference on Machine Learning},
  pages={2417--2426},
  year={2016},
  organization={PMLR}
}


@inproceedings{amos2017input,
  title={Input convex neural networks},
  author={Amos, Brandon and Xu, Lei and Kolter, J Zico},
  booktitle={International Conference on Machine Learning},
  pages={146--155},
  year={2017},
  organization={PMLR}
}


@article{hanna2009direct,
  title={Direct cell reprogramming is a stochastic process amenable to acceleration},
  author={Hanna, Jacob and Saha, Krishanu and Pando, Bernardo and Van Zon, Jeroen and Lengner, Christopher J and Creyghton, Menno P and van Oudenaarden, Alexander and Jaenisch, Rudolf},
  journal={Nature},
  volume={462},
  number={7273},
  pages={595--601},
  year={2009},
  publisher={Nature Publishing Group UK London}
}


@article{morris2014mathematical,
  title={Mathematical approaches to modeling development and reprogramming},
  author={Morris, Rob and Sancho-Martinez, Ignacio and Sharpee, Tatyana O and Izpisua Belmonte, Juan Carlos},
  journal={Proceedings of the National Academy of Sciences},
  volume={111},
  number={14},
  pages={5076--5082},
  year={2014},
  publisher={National Acad Sciences}
}


@article{yang2022diffusion,
  title={Diffusion models: A comprehensive survey of methods and applications},
  author={Yang, Ling and Zhang, Zhilong and Song, Yang and Hong, Shenda and Xu, Runsheng and Zhao, Yue and Zhang, Wentao and Cui, Bin and Yang, Ming-Hsuan},
  journal={ACM Computing Surveys},
  year={2022},
  publisher={ACM New York, NY}
}


@article{Lanzetti2022, 
title = {First-order Conditions for Optimization in the {W}asserstein Space}, 
author = {Lanzetti, Nicolas and Bolognani, Saverio and D{\"o}rfler, Florian}, 
journal = {SIAM Journal on Mathematics of Data Science}, 
year = {2024},
note={In press},
}


@article{santambrogio2015optimal,
  title={Optimal transport for applied mathematicians},
  author={Santambrogio, Filippo},
  journal={Birk{\"a}user, NY},
  volume={55},
  number={58-63},
  pages={94},
  year={2015},
  publisher={Springer}
}

@article{luo2023improved,
  title={Improved Complexity Analysis of the {S}inkhorn and {G}reenkhorn Algorithms for Optimal Transport},
  author={Luo, Jianzhou and Yang, Dingchuan and Wei, Ke},
  journal={arXiv preprint arXiv:2305.14939},
  year={2023}
}

@inproceedings{kent2021frank,
 author = {Kent, Carson and Li, Jiajin and Blanchet, Jose and Glynn, Peter W},
 booktitle = {Advances in Neural Information Processing Systems},
 pages = {14448--14462},
 publisher = {Curran Associates, Inc.},
 title = {Modified {F}rank {W}olfe in Probability Space},
 volume = {34},
 year = {2021}
}

@inproceedings{esser2024scaling,
  title={Scaling rectified flow transformers for high-resolution image synthesis},
  author={Esser, Patrick and Kulal, Sumith and Blattmann, Andreas and Entezari, Rahim and M{\"u}ller, Jonas and Saini, Harry and Levi, Yam and Lorenz, Dominik and Sauer, Axel and Boesel, Frederic and others},
  booktitle={Forty-first International Conference on Machine Learning},
  year={2024}
}

@article{burger2009mixed,
  title={A mixed finite element method for nonlinear diffusion equations},
  author={Burger, Martin and Carrillo de la Plata, Jos{\'e} Antonio and Wolfram, Marie-Therese},
  year={2009},
  publisher={Centre de Recerca Matem{\`a}tica}
}


@article{benamou2016augmented,
  title={An augmented {L}agrangian approach to {W}asserstein gradient flows and applications},
  author={Benamou, Jean-David and Carlier, Guillaume and Laborde, Maxime},
  journal={ESAIM: Proceedings and surveys},
  volume={54},
  pages={1--17},
  year={2016},
  publisher={EDP Sciences}
}


@article{peyre2015entropic,
  title={Entropic approximation of {W}asserstein gradient flows},
  author={Peyr{\'e}, Gabriel},
  journal={SIAM Journal on Imaging Sciences},
  volume={8},
  number={4},
  pages={2323--2351},
  year={2015},
  publisher={SIAM}
}


@article{carrillo2022primal,
  title={Primal dual methods for {W}asserstein gradient flows},
  author={Carrillo, Jos{\'e} A and Craig, Katy and Wang, Li and Wei, Chaozhen},
  journal={Foundations of Computational Mathematics},
  pages={1--55},
  year={2022},
  publisher={Springer}
}

@article{erbe2023transcriptomic,
  title={Transcriptomic forecasting with neural ordinary differential equations},
  author={Erbe, Rossin and Stein-O’Brien, Genevieve and Fertig, Elana J},
  journal={Patterns},
  volume={4},
  number={8},
  year={2023},
  publisher={Elsevier}
}


@article{fisher2009data,
  title={Data assimilation in weather forecasting: a case study in PDE-constrained optimization},
  author={Fisher, Mike and Nocedal, Jorge and Tr{\'e}molet, Yannick and Wright, Stephen J},
  journal={Optimization and Engineering},
  volume={10},
  number={3},
  pages={409--426},
  year={2009},
  publisher={Springer}
}

@article{moon2019visualizing,
  title={Visualizing structure and transitions in high-dimensional biological data},
  author={Moon, Kevin R and van Dijk, David and Wang, Zheng and Gigante, Scott and Burkhardt, Daniel B and Chen, William S and Yim, Kristina and Elzen, Antonia van den and Hirn, Matthew J and Coifman, Ronald R and others},
  journal={Nature biotechnology},
  volume={37},
  number={12},
  pages={1482--1492},
  year={2019},
  publisher={Nature Publishing Group US New York}
}


@article{mokrov2021large,
  title={Large-scale {W}asserstein gradient flows},
  author={Mokrov, Petr and Korotin, Alexander and Li, Lingxiao and Genevay, Aude and Solomon, Justin M and Burnaev, Evgeny},
  journal={Advances in Neural Information Processing Systems},
  volume={34},
  pages={15243--15256},
  year={2021}
}

@misc{tange_2024_13826092,
  author       = {Tange, Ole},
  title        = {GNU Parallel},
  month        = sep,
  year         = 2024,
  publisher    = {Zenodo},
}

@article{RNA_dataset,
  title={Visualizing structure and transitions in high-dimensional biological data},
  author={Moon, Kevin R and Van Dijk, David and Wang, Zheng and Gigante, Scott and Burkhardt, Daniel B and Chen, William S and Yim, Kristina and Elzen, Antonia van den and Hirn, Matthew J and Coifman, Ronald R and others},
  journal={Nature biotechnology},
  volume={37},
  number={12},
  pages={1482--1492},
  year={2019},
  publisher={Nature Publishing Group US New York}
}
