simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-space (by empirical/analytical NTK).
☆16Oct 21, 2020Updated 5 years ago
Alternatives and similar repositories for ngd_in_wide_nn
Users that are interested in ngd_in_wide_nn are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Improving predictions of Bayesian neural nets via local linearization, AISTATS 2021☆15Dec 30, 2022Updated 3 years ago
- SKFAC Preconditioner for MindSpore☆12Jul 2, 2021Updated 5 years ago
- Code for Knowledge-Adaptation Priors based on the NeurIPS 2021 paper by Khan and Swaroop.☆17Feb 1, 2022Updated 4 years ago
- Pytorch implementation of KFAC - this is a port of https://github.com/tensorflow/kfac/☆32Jun 6, 2024Updated 2 years ago
- A toolbox for trajectory optimization of dynamical systems☆56Jun 16, 2022Updated 4 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Computes trajectories for evolutionary dynamics.☆15Oct 6, 2020Updated 5 years ago
- Simple and extensible hypergradient for PyTorch☆18Feb 23, 2023Updated 3 years ago
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆25Mar 4, 2022Updated 4 years ago
- "Detecting Extrapolation with Local Ensembles" by David Madras, James Atwood, and Alex D'Amour☆13Sep 25, 2020Updated 5 years ago
- Approximate Inference Turns Deep Networks into Gaussian Processes (dnn2gp)☆48Dec 17, 2019Updated 6 years ago
- Dirichlet Process Mixture Models in Julia☆14May 31, 2016Updated 10 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆29Aug 5, 2025Updated 11 months ago
- Generalised UDRL☆37May 12, 2022Updated 4 years ago
- Code for Accelerated Linearized Laplace Approximation for Bayesian Deep Learning (ELLA, NeurIPS 22')☆17Oct 12, 2022Updated 3 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Measurements of Three-Level Hierarchical Structure in the Outliers in the Spectrum of Deepnet Hessians (ICML 2019)☆16Apr 27, 2019Updated 7 years ago
- Variational Inference by Policy Search☆13Apr 24, 2019Updated 7 years ago
- Code for the paper☆11May 24, 2024Updated 2 years ago
- The fast Finite Volume simulator with UQ support.☆28Feb 16, 2025Updated last year
- Implementation of different Relative Entropy Policy Search flavors☆13Nov 15, 2021Updated 4 years ago
- A simple Jax implementation of influence functions.☆21Apr 9, 2024Updated 2 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
- Code repository for the paper "Adaptive Bounding Box Uncertainties via Two-Step Conformal Prediction" @ ECCV 2024 (Oral)☆13Apr 22, 2025Updated last year
- A toolbox for inference of mixture models☆16May 29, 2023Updated 3 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Mirror of nofib repository. DO NOT SUBMIT PULL REQUESTS HERE☆14Sep 19, 2019Updated 6 years ago
- ☆12Nov 9, 2020Updated 5 years ago
- Estimators for Information Theoretic Functionals using Influence Functions☆11Apr 17, 2016Updated 10 years ago
- Official code for Coupled Oscillatory RNN (ICLR 2021, Oral)☆53Aug 26, 2021Updated 4 years ago
- Build and run container environment for LFRic☆11Jan 8, 2024Updated 2 years ago
- ☆14Jul 22, 2021Updated 4 years ago
- Working on this project as part of 3D-Vision course at ETH☆11Jul 20, 2020Updated 5 years ago
- Pytorch2Jax is a small Python library that provides functions that wraps PyTorch models into Jax functions and Flax modules.☆21Feb 20, 2023Updated 3 years ago
- ☆18Apr 11, 2023Updated 3 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- The project page of paper: Architecture disentanglement for deep neural networks [ICCV 2021, oral]☆16Nov 27, 2021Updated 4 years ago
- The Code for Homeomorphic Projection☆14Sep 21, 2023Updated 2 years ago
- E2C implementation in PyTorch☆43Jul 5, 2017Updated 9 years ago
- ☆10Mar 25, 2024Updated 2 years ago
- Code for "Functional variational Bayesian neural networks" (https://arxiv.org/abs/1903.05779)☆88Jul 19, 2020Updated 5 years ago
- Implementation of Paper: Long-term Forecasting with TiDE: Time-series Dense Encoder☆21Nov 1, 2024Updated last year
- Code for reproducing experiments performed for Accoridon☆13Jun 11, 2021Updated 5 years ago