simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-space (by empirical/analytical NTK).
☆17Oct 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☆16Dec 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/☆33Jun 6, 2024Updated 2 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
- 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.
- 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)☆49Dec 17, 2019Updated 6 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆29Aug 5, 2025Updated last year
- The fast Finite Volume simulator with UQ support.☆28Feb 16, 2025Updated last year
- Code for Accelerated Linearized Laplace Approximation for Bayesian Deep Learning (ELLA, NeurIPS 22')☆17Oct 12, 2022Updated 3 years ago
- 심층강화학습 책 https://hiddenbeginner.github.io/Deep-Reinforcement-Learnings☆11May 10, 2024Updated 2 years ago
- 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.
- We define and estimate smooth unique information of samples with respect to classifier weights and predictions. We compute these quantiti…☆11Mar 9, 2021Updated 5 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 for the paper☆11May 24, 2024Updated 2 years ago
- Implementation of different Relative Entropy Policy Search flavors☆13Nov 15, 2021Updated 4 years ago
- HCOMP '22 -- Eliciting and Learning with Soft Labels from Every Annotator☆11Nov 1, 2022Updated 3 years ago
- A simple Jax implementation of influence functions.☆21Apr 9, 2024Updated 2 years ago
- Implementation of the skill discovery algorithm described in ICLR submission "Option Discovery using Deep Skill Chaining"☆30Sep 24, 2019Updated 6 years ago
- Code repository for the paper "Adaptive Bounding Box Uncertainties via Two-Step Conformal Prediction" @ ECCV 2024 (Oral)☆13Apr 22, 2025Updated last year
- Official code for AL-PINNS: Augmented Lagrangian relaxation method for Physics-Informed Neural Networks☆12Jul 29, 2023Updated 3 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- ☆14Aug 22, 2021Updated 4 years ago
- Python bindings to some optimization benchmarks (robotics problems), in order to constrained optimization solvers. Includes also an inter…☆21Dec 8, 2022Updated 3 years ago
- ☆11Jun 17, 2016Updated 10 years ago
- ☆62Jun 8, 2021Updated 5 years ago
- ☆12Nov 9, 2020Updated 5 years ago
- PLASMA parallel library for dense linear algebra.☆10May 30, 2017Updated 9 years ago
- Build and run container environment for LFRic☆11Jan 8, 2024Updated 2 years ago
- Variance Gamma distribution (Python): pdf, cdf, rand and fit.☆11Mar 8, 2018Updated 8 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
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Official codebase for Generating Diverse Cooperative Agents by Learning Incompatible Policies (notable-top-25% @ ICLR 2023)☆19May 10, 2024Updated 2 years ago
- ☆17Apr 11, 2023Updated 3 years ago
- High Performance Computing for Weather and Climate☆45Jun 26, 2026Updated last month
- The Code for Homeomorphic Projection☆14Sep 21, 2023Updated 2 years ago
- A lightweight library for mDNS swarm discovery with bounded bandwidth usage☆21Jul 13, 2026Updated last month
- Code for "Functional variational Bayesian neural networks" (https://arxiv.org/abs/1903.05779)☆88Jul 19, 2020Updated 6 years ago
- ☆10Mar 25, 2024Updated 2 years ago