Fast and Easy Infinite Neural Networks in Python

[ICML2022] Variational Wasserstein gradient flow

Pytorch implementation of DGflow (ICLR 2021).

Numeric simulation of a 2D Bose Einstein condensate

[NeurIPS 2022] Old can be Gold: Better Gradient Flow can Make Vanilla-GCNs Great Again by Ajay Jaiswal*, Peihao Wang*, Tianlong Chen, Justin F Rousseau, Ying Ding, Zhangyang Wang

Solver for differential algebraic equations

Variational Filtering via Wasserstein Gradient Flow

Senior Project for Statistics & Data Science at Yale University

Discretized Wasserstein Particle Flows of a MMD-regularized f-divergence functional.

Yale S&DS 432 final project studying lazy training dynamics for differentiable optimization problems

Code for paper: "Improved Residual Network Based on Norm-Preservation for Visual Recognition" https://doi.org/10.1016/j.neunet.2022.10.023

Accelerated Stein Variational Gradient Descent for sampling for densities

Supplementary code for the paper 'Are GATs Out of Balance?' to be published at NeurIPS 2023

Discrete approximation to 3D winding number mapping T^3->U(N)

Utilities of quantum theory of (higher-form) general gauge fields