generativeai-tue / vsgdLinks
Code repository of the paper "Variational Stochastic Gradient Descent for Deep Neural Networks" published at
☆41Updated 3 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☆164Updated last week
- Running Jax in PyTorch Lightning☆113Updated 8 months ago
- Uncertainty quantification with PyTorch☆372Updated 4 months ago
- Graph neural networks in JAX.☆67Updated last year
- The boundary of neural network trainability is fractal☆215Updated last year
- Materials of the Nordic Probabilistic AI School 2022.☆180Updated 2 years ago
- A projection-based framework for gradient-free and parallel learning☆104Updated 2 months ago
- Diffusion models in PyTorch☆107Updated 2 months ago
- ☆125Updated 8 months ago
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆73Updated 2 years ago
- Agustinus' very opiniated publication-ready plotting library☆69Updated 3 months ago
- TorchFSM: Fourier Spectral Method with PyTorch☆48Updated 3 weeks ago
- High order and sparse layers in pytorch. Lagrange Polynomial, Piecewise Lagrange Polynomial, Piecewise Discontinuous Lagrange Polynomial…☆45Updated last year
- ☆150Updated last year
- Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.☆172Updated 2 years ago
- Lightning-like training API for JAX with Flax☆42Updated 8 months ago
- ☆30Updated last month
- Deep Learning, an Energy Approach☆204Updated 2 months ago
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- A library for uncertainty quantification based on PyTorch☆122Updated 3 years ago
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆308Updated 2 months ago
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆129Updated 3 years ago
- Generating and Imputing Tabular Data via Diffusion and Flow XGBoost Models☆162Updated last year
- Code for the book "The Elements of Differentiable Programming".☆260Updated 2 months ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆133Updated last year
- ☆115Updated last week
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆99Updated 2 months ago
- Classfication metrics and post-hoc calibration☆32Updated 2 months ago
- Topological Deep Learning☆292Updated this week