Sinkhorn Barycenter via Functional Gradient Descent. - Zebang Shen; Zhenfu Wang; Alejandro Ribeiro; Hamed Hassani (NeurIPS, 2020)

Sinkhorn Natural Gradient for Generative Models. - Zebang Shen; Zhenfu Wang; Alejandro Ribeiro; Hamed Hassani (NeurIPS, 2020)

Self-Consistency of the Fokker Planck Equation. - Zebang Shen; Zhenfu Wang; Satyen Kale; Alejandro Ribeiro; Amin Karbasi; Hamed Hassani (COLT, 2022)

Entropy-dissipation Informed Neural Network for McKean-Vlasov Type PDEs. - Zebang Shen; Zhenfu Wang (NeurIPS, 2023)

Transport based particle methods for the Fokker-Planck-Landau equation. - Vasily Ilin; Jingwei Hu; Zhenfu Wang (CoRR, 2024)