AISTATS 2025: Computing high-dimensional optimal transport by flow neural networks

Official implementation of "Local Flow Matching Generative Models"

Official implementation of "Normalizing flow neural networks by JKO scheme" (NeurIPS 2023 spotlight)

Official implementation of the ICML 2023 work "Sequential Predictive Conformal Inference"

Official implementation of "Conformal prediction for multi-dimensional time series by ellipsoidal sets" (ICML 2024 spotlight)

Implementation for "An alternative approach to train neural networks using monotone variational inequality"

Official implementation of "Flow-based Distributionally Robust Optimization"

Official code for: Conformal prediction interval for dynamic time-series (conference, ICML 21 Long Presentation) AND Conformal prediction for time-series (journal, IEEE TPAMI)

Tutorials on training GNN models

Official code for the paper: Invertible Neural Network for Graph Prediction

Official repository for the ICML 2022 DFUQ paper: conformal prediction sets for time-series