3 papers
cs.LG2026
FlashSinkhorn: IO-Aware Entropic Optimal Transport on GPU
Felix X. -F. Ye, Xingjie Li, An Yu +3
Entropic optimal transport (EOT) via Sinkhorn iterations is widely used in modern machine learning, yet GPU solvers remain inefficient at scale. Tensorized implementations suffer q…
cs.DC2024
Modeling and Simulation of Traffic on I-485 via Linear Systems and Iterative Methods
Dominic Kealoha, Fabiola Rojas, Xingjie Li
Iterative methods such as Jacobi, Gauss-Seidel, and Successive Over-Relaxation (SOR) are fundamental tools in solving large systems of linear equations across various scientific fi…
math.NA2024
Robust First and Second-Order Differentiation for Regularized Optimal Transport
Xingjie Li, Fei Lu, Molei Tao +1
Applications such as unbalanced and fully shuffled regression can be approached by optimizing regularized optimal transport (OT) distances, such as the entropic OT and Sinkhorn dis…