kazukiosawa / ngd_in_wide_nnView external linksLinks
simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-space (by empirical/analytical NTK).
☆15Oct 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
Sorting:
- Pytorch implementation of KFAC - this is a port of https://github.com/tensorflow/kfac/☆30Jun 6, 2024Updated last year
- Simple and extensible hypergradient for PyTorch☆18Feb 23, 2023Updated 2 years ago
- Improving predictions of Bayesian neural nets via local linearization, AISTATS 2021☆16Dec 30, 2022Updated 3 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆28Aug 5, 2025Updated 6 months ago
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆25Mar 4, 2022Updated 3 years ago
- A toolbox for trajectory optimization of dynamical systems☆55Jun 16, 2022Updated 3 years ago
- Implementation of the skill discovery algorithm described in ICLR submission "Option Discovery using Deep Skill Chaining"☆30Sep 24, 2019Updated 6 years ago
- This repo contains all the code related to the interactive online Bloch sphere simulator☆18Jan 19, 2026Updated 3 weeks ago
- Contains the code of estimating cloud data center workload with ARIMA, SARIMA, LSTM, RNN.☆12Nov 1, 2021Updated 4 years ago
- High Performance Computing for Weather and Climate☆42Feb 3, 2026Updated 2 weeks ago
- Generalised UDRL☆37May 12, 2022Updated 3 years ago
- ☆33May 21, 2020Updated 5 years ago
- Build and run container environment for LFRic☆10Jan 8, 2024Updated 2 years ago
- A data processing module implemented with numpy☆10Aug 16, 2022Updated 3 years ago
- Distributed Communication-Optimal Shuffle and Transpose Algorithm☆14Feb 4, 2026Updated last week
- The Code for Homeomorphic Projection☆13Sep 21, 2023Updated 2 years ago
- Official code for AL-PINNS: Augmented Lagrangian relaxation method for Physics-Informed Neural Networks☆12Jul 29, 2023Updated 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
- A PyTorch implementation of WRN☆10Jun 2, 2018Updated 7 years ago
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆11May 8, 2023Updated 2 years ago
- Attentional Mechanism incorporated in Asynchronous Advantage Actor Critic a3c/a2c deep mind☆10Jan 9, 2018Updated 8 years ago
- ☆11Nov 9, 2020Updated 5 years ago
- High-Level Training, Data Augmentation, and Utilities for Pytorch☆13Mar 8, 2019Updated 6 years ago
- Data Flow Matrix Machines. Generalization of recurrent neural networks.☆15Dec 24, 2024Updated last year
- Fast interpolative decompositions in Python☆10Jan 4, 2021Updated 5 years ago
- ☆10Feb 19, 2019Updated 6 years ago
- Offline Policy Evaluation via Adaptive Weighting with Data from Contextual Bandits☆10Oct 21, 2024Updated last year
- Some commonly used functions and modules☆10Jan 15, 2024Updated 2 years ago
- Multi-Objective Causal Bayesian Optimisation, a new paradigm for finding Pareto-optimal interventions in multi-outcome causal models☆17Jun 2, 2025Updated 8 months ago
- A Type System Attempt for Wolfram Language☆13Feb 28, 2021Updated 4 years ago
- PDBe Programming Interfaces information☆14Jul 23, 2018Updated 7 years ago
- Computes trajectories for evolutionary dynamics.☆15Oct 6, 2020Updated 5 years ago
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 years ago
- JAX/Haiku implementation of "Auction Learning as a Two-Player Game"☆11Jul 6, 2024Updated last year
- Hyperoctree construction and manipulation☆11Jan 4, 2021Updated 5 years ago
- Model-based time series clustering using variational inference.☆12Oct 28, 2018Updated 7 years ago
- Mis proyectos de marketing aplicando AI☆11Oct 31, 2025Updated 3 months ago
- yet another reinforcement learning package☆12May 24, 2022Updated 3 years ago
- ☆10Mar 25, 2024Updated last year