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 6 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
- ☆12Oct 17, 2018Updated 8 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.☆16Feb 1, 2022Updated 4 years ago
- [NeurIPS 2024] Codes for "Understanding Linear Probing then Fine-tuning Language Models from NTK Perspective"☆11Dec 22, 2024Updated last year
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- 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
- An Optax-based JAX implementation of the IVON optimizer for large-scale VI training of NNs (ICML'24 spotlight)☆10Dec 19, 2024Updated last year
- Approximate Inference Turns Deep Networks into Gaussian Processes (dnn2gp)☆49Dec 17, 2019Updated 6 years ago
- Implementation for the paper "Approximating full conformal prediction at scale via influence functions"☆11Apr 25, 2024Updated 2 years ago
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆25Mar 4, 2022Updated 4 years ago
- Dirichlet Process Mixture Models in Julia☆14May 31, 2016Updated 10 years ago
- Generalised UDRL☆37May 12, 2022Updated 4 years ago
- ☆13Aug 15, 2024Updated 2 years ago
- Simple and extensible hypergradient for PyTorch☆18Feb 23, 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.
- 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
- The fast Finite Volume simulator with UQ support.☆30Feb 16, 2025Updated last year
- 심층강화학습 책 https://hiddenbeginner.github.io/Deep-Reinforcement-Learnings☆11May 10, 2024Updated 2 years ago
- Variational Inference by Policy Search☆13Apr 24, 2019Updated 7 years ago
- ☆10Sep 12, 2021Updated 5 years ago
- Deep Reinforcement Learning Toolbox for Robotics using Keras and TensorFlow☆19Aug 16, 2019Updated 7 years ago
- Code for the paper☆12May 24, 2024Updated 2 years ago
- A toolbox for inference of mixture models☆16May 29, 2023Updated 3 years ago
- ☆17Jun 10, 2024Updated 2 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.
- HCOMP '22 -- Eliciting and Learning with Soft Labels from Every Annotator☆11Nov 1, 2022Updated 4 years ago
- Model-based reinforcement learning using CEM, MPC and PETS☆15Nov 20, 2019Updated 6 years ago
- implementation of Wasserstein Natural Policy Gradients and Wasserstein Natural Evolution Strategies☆13Mar 9, 2021Updated 5 years ago
- Implementation of the skill discovery algorithm described in ICLR submission "Option Discovery using Deep Skill Chaining"☆30Sep 24, 2019Updated 7 years ago
- Code to reproduce the experiments in The Mirage of Action-Dependent Baselines in Reinforcement Learning.☆17Aug 2, 2018Updated 8 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
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆332Oct 13, 2026Updated 3 weeks ago
- ☆62Jun 8, 2021Updated 5 years ago
- Hyperoctree construction and manipulation☆12Jan 4, 2021Updated 5 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.
- ☆14Jul 22, 2021Updated 5 years ago
- PLASMA parallel library for dense linear algebra.☆10May 30, 2017Updated 9 years ago
- Implementation of PILCO for the Model-Based Baselines Project☆18Jul 18, 2019Updated 7 years ago
- Estimators for Information Theoretic Functionals using Influence Functions☆11Apr 17, 2016Updated 10 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
- PyLIS (Life in Silico with PyBullet)☆18May 5, 2022Updated 4 years ago
- ☆17Apr 11, 2023Updated 3 years ago