hendrydong / NTK-and-MF-examples
Visualization of mean field and neural tangent kernel regime
☆20Updated 3 months ago
Related projects ⓘ
Alternatives and complementary repositories for NTK-and-MF-examples
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆62Updated 2 years ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆43Updated last year
- ☆31Updated 4 years ago
- ☆23Updated 3 years ago
- Monotone operator equilibrium networks☆51Updated 4 years ago
- Towards Understanding Sharpness-Aware Minimization [ICML 2022]☆35Updated 2 years ago
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆22Updated 11 months ago
- ☆67Updated 5 years ago
- Efficient Riemannian Optimization on Stiefel Manifold via Cayley Transform☆36Updated 5 years ago
- [NeurIPS 2021] A Geometric Analysis of Neural Collapse with Unconstrained Features☆53Updated 2 years ago
- ☆15Updated last year
- Training vision models with full-batch gradient descent and regularization☆38Updated last year
- ☆57Updated last year
- ☆59Updated 3 years ago
- Source code of "What can linearized neural networks actually say about generalization?☆18Updated 3 years ago
- [ICML 2021] This is the official github repo for training L_inf dist nets with high certified accuracy.☆41Updated 2 years ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆26Updated 3 years ago
- An Elegant Library for Bayesian Deep Learning in PyTorch☆22Updated last year
- ☆40Updated last year
- ☆49Updated 3 years ago
- Code for "Implicit Normalizing Flows" (ICLR 2021 spotlight)☆34Updated 3 years ago
- ☆25Updated last year
- ☆33Updated 4 years ago
- Euclidean Wasserstein-2 optimal transportation☆44Updated last year
- Sinkhorn Barycenters via Frank-Wolfe algorithm☆24Updated 4 years ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆97Updated 4 years ago
- ☆23Updated 2 years ago
- ☆21Updated last year
- Code for experiments in my blog post on the Neural Tangent Kernel: https://eigentales.com/NTK☆166Updated 5 years ago