An Optax-based JAX implementation of the IVON optimizer for large-scale VI training of NNs (ICML'24 spotlight)
☆10Dec 19, 2024Updated last year
Alternatives and similar repositories for ivon-optax
Users that are interested in ivon-optax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Code for "SAM as an Optimal Relaxation of Bayes", ICLR 2023.☆26Jul 21, 2023Updated 3 years ago
- IVON optimizer for neural networks based on variational learning.☆93Nov 7, 2024Updated last year
- Code for Knowledge-Adaptation Priors based on the NeurIPS 2021 paper by Khan and Swaroop.☆16Feb 1, 2022Updated 4 years ago
- Contains code for the NeurIPS 2020 paper by Pan et al., "Continual Deep Learning by FunctionalRegularisation of Memorable Past"☆44Nov 10, 2020Updated 5 years ago
- Expandable And Scalable Infrastructure for Finite Element Methods, EASIFEM, is [Modern Fortran](https://fortran-lang.org) framework for s…☆12Updated this week
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- ☆13Oct 15, 2023Updated 3 years ago
- Tensor-Train toolbox in Jax☆12Sep 27, 2021Updated 5 years ago
- Distributed Deep Learning Framework for Web☆18Jul 29, 2026Updated 3 months ago
- LaTeX style file for Transactions on Machine Learning Research☆15Jun 30, 2023Updated 3 years ago
- Code base to generate evidence map of natural climate solutions related scientific papers☆13Nov 22, 2024Updated last year
- Dirichlet Process Mixture Models in Julia☆14May 31, 2016Updated 10 years ago
- Repository for the paper "Bayesian Quadrature on Riemannian Data Manifolds"☆10Jun 18, 2024Updated 2 years ago
- ☆19Oct 16, 2026Updated 2 weeks ago
- Posterior Server☆16Nov 10, 2016Updated 9 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Regression datasets from the UCI repository with standardized test-train splits.☆55Jul 19, 2022Updated 4 years ago
- An example showing how to use jax to train resnet50 on multi-node multi-GPU☆20Jul 4, 2022Updated 4 years ago
- A package for computing matrix exponentials and finite horizon Gramians☆11Jan 21, 2026Updated 9 months ago
- Implementation of Deep Learning Models in JAX (equinox)☆25Oct 21, 2026Updated last week
- PyTorch implementation of paper "Flat Metric Minimization with Applications in Generative Modeling"☆19May 14, 2019Updated 7 years ago
- A simple Jax implementation of influence functions.☆21Apr 9, 2024Updated 2 years ago
- Make Rmarkdown and Shiny apps fullscreen☆10Sep 1, 2021Updated 5 years ago
- ☆13Aug 15, 2024Updated 2 years ago
- Code accompanying the NeurIPS 2021 Paper: A Probabilistic State Space Model for Joint Inference from Differential Equations and Data (Sch…☆13Nov 7, 2022Updated 3 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Code for sublabel-accurate multi-labeling papers (published at CVPR '16, ECCV '16)☆20Oct 11, 2016Updated 10 years ago
- Variational Inference by Policy Search☆13Apr 24, 2019Updated 7 years ago
- Convolutions and more as einsum for PyTorch☆18Jun 6, 2024Updated 2 years ago
- Exercises for the Tutorial on Approximate Bayesian Inference at the Data Science Summer School 2018☆22Aug 6, 2018Updated 8 years ago
- This repository collects supplementary material to study reinforcement learning with a focus on topics covered by the TU Darmstadt IAS le…☆12Feb 11, 2019Updated 7 years ago
- Implementation of different Relative Entropy Policy Search flavors☆13Nov 15, 2021Updated 4 years ago
- simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…☆17Oct 21, 2020Updated 6 years ago
- Systematic Review Query Visualisation and Understanding Interface☆17Dec 5, 2025Updated 10 months ago
- ☆54Sep 16, 2026Updated last month
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Code for AISTATS 2017 paper on "Conjugate-Computation Variational Inference"☆20Jul 10, 2020Updated 6 years ago
- Deep Reinforcement Learning Toolbox for Robotics using Keras and TensorFlow☆19Aug 16, 2019Updated 7 years ago
- A toolbox for inference of mixture models☆16May 29, 2023Updated 3 years ago
- A generic library for linear and non-linear Gaussian smoothing problems. The code leverages JAX and implements several linearization algo…☆14Apr 20, 2026Updated 6 months ago
- ☆18Sep 27, 2023Updated 3 years ago
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆15May 8, 2023Updated 3 years ago
- Example code to speed up GP inference with gradients for high-dimensional inputs.☆14Jul 12, 2021Updated 5 years ago