generativeai-tue / vsgdLinks
Code repository of the paper "Variational Stochastic Gradient Descent for Deep Neural Networks" published at
☆38Updated last month
Alternatives and similar repositories for vsgd
Users that are interested in vsgd are comparing it to the libraries listed below
Sorting:
- High order and sparse layers in pytorch. Lagrange Polynomial, Piecewise Lagrange Polynomial, Piecewise Discontinuous Lagrange Polynomial…☆44Updated 11 months ago
- TopoBench is a Python library designed to standardize benchmarking and accelerate research in Topological Deep Learning☆120Updated this week
- Graph neural networks in JAX.☆67Updated 11 months ago
- ☆150Updated 9 months ago
- Running Jax in PyTorch Lightning☆102Updated 5 months ago
- A State-Space Model with Rational Transfer Function Representation.☆78Updated last year
- ☆48Updated 4 months ago
- This repository contains a better implementation of Kolmogorov-Arnold networks☆61Updated this week
- ☆114Updated this week
- Recursive Bayesian Estimation (Sequential / Online Inference)☆58Updated last year
- Lightning-like training API for JAX with Flax☆38Updated 5 months ago
- Efficiently Composable Data Augmentation on the GPU with Jax☆35Updated 3 weeks ago
- Exca - Execution and caching tool for python☆85Updated this week
- TorchFSM: Fourier Spectral Method with PyTorch☆43Updated this week
- PyHopper is a hyperparameter optimizer, made specifically for high-dimensional problems arising in machine learning research.☆86Updated last year
- Patched Attention for Nonlinear Dynamics☆132Updated last week
- Kolmogorov–Arnold Networks with modified activation (using MLP to represent the activation)☆105Updated 7 months ago
- Diffusion models in PyTorch☆102Updated this week
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆72Updated 2 years ago
- Gradient-based constrained optimization for JAX☆32Updated 2 years ago
- TorchDR - PyTorch Dimensionality Reduction☆115Updated last week
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆27Updated last month
- Reconstructing shared causal drivers from noisy time series☆53Updated 7 months ago
- Generating and Imputing Tabular Data via Diffusion and Flow XGBoost Models☆155Updated 10 months ago
- A system for automating selection and optimization of pre-trained models from the TAO Model Zoo☆25Updated 11 months ago
- Materials of the Nordic Probabilistic AI School 2023.☆90Updated last year
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆48Updated 2 months ago
- Pytorch-like dataloaders for JAX.☆83Updated last week
- A metrics library for the JAX ecosystem☆40Updated 2 years ago
- Kolmogorov-Arnold Networks with various basis functions like B-Splines, Fourier, Chebyshev, Wavelets etc☆35Updated last year