generativeai-tue / vsgdLinks
Code repository of the paper "Variational Stochastic Gradient Descent for Deep Neural Networks" published at
☆41Updated 7 months ago
Alternatives and similar repositories for vsgd
Users that are interested in vsgd are comparing it to the libraries listed below
Sorting:
- TopoBench is a Python library designed to standardize benchmarking and accelerate research in Topological Deep Learning☆215Updated this week
- Diffusion models in PyTorch☆120Updated 2 weeks ago
- ☆129Updated last year
- Uncertainty quantification with PyTorch☆377Updated 2 months ago
- Lightning-like training API for JAX with Flax☆44Updated last year
- Running Jax in PyTorch Lightning☆114Updated last year
- Graph neural networks in JAX.☆68Updated last year
- A library for uncertainty quantification based on PyTorch☆122Updated 3 years ago
- Geometric kernels on manifolds, meshes and graphs☆281Updated this week
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆311Updated 5 months ago
- The boundary of neural network trainability is fractal☆221Updated last year
- PyTorch implementation of Levenberg-Marquardt training algorithm☆88Updated 3 weeks ago
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆50Updated 5 months ago
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆140Updated 2 years ago
- A parallel ODE solver for PyTorch☆273Updated last year
- Clifford-Steerable Convolutional Neural Networks [ICML'24]☆51Updated 7 months ago
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆30Updated 9 months ago
- TorchFSM: Fourier Spectral Method with PyTorch☆53Updated 2 weeks ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆147Updated last month
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆108Updated last year
- Bayesian algorithm execution (BAX)☆55Updated 4 years ago
- Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…☆275Updated 3 years ago
- A projection-based framework for gradient-free and parallel learning☆108Updated 6 months ago
- code for "Universal Functional Regression with Neural Operator Flows" TMLR 2024☆19Updated 6 months ago
- Library to make any existing neural network architecture equivariant☆56Updated last year
- Normalizing flows in PyTorch☆433Updated 2 weeks ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆188Updated last week
- ☆210Updated last year
- ☆153Updated last month