gradient-flow topic

List gradient-flow repositories

neural-tangents

2.2k
Stars
227
Forks
Watchers

Fast and Easy Infinite Neural Networks in Python

verified publisher icon google

DGflow

17
Stars
5
Forks
Watchers

Pytorch implementation of DGflow (ICLR 2021).

verified publisher icon clear-nus

variational_wgf

20
Stars
3
Forks
Watchers

[ICML2022] Variational Wasserstein gradient flow

verified publisher icon sbyebss