tensorflow / kfacLinks
An implementation of KFAC for TensorFlow
☆198Updated 3 years ago
Alternatives and similar repositories for kfac
Users that are interested in kfac are comparing it to the libraries listed below
Sorting:
- ☆133Updated 7 years ago
- Hypergradient descent☆149Updated last year
- hessian in pytorch☆187Updated 4 years ago
- Implementation of Model-Agnostic Meta-Learning (MAML) in Jax☆191Updated 2 years ago
- A colab that implements the Symplectic Gradient Adjustment optimizer from "The mechanics of n-player differentiable games"☆153Updated 6 years ago
- Guided Evolutionary Strategies☆271Updated 2 years ago
- Implementation of Spectral Inference Networks, ICLR 2019☆172Updated 6 years ago
- Concise deep learning for JAX☆183Updated 4 years ago
- Hessian spectral density estimation in TF and Jax☆123Updated 4 years ago
- ☆169Updated 11 months ago
- Optimizing control variates for black-box gradient estimation☆163Updated 5 years ago
- Experiments for the paper "Exponential expressivity in deep neural networks through transient chaos"☆71Updated 9 years ago
- Autoregressive Energy Machines☆78Updated 2 years ago
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆143Updated 2 years ago
- Neural Architecture Search with Bayesian Optimisation and Optimal Transport☆134Updated 6 years ago
- ☆219Updated 7 years ago
- Forward-mode Automatic Differentiation for TensorFlow☆139Updated 7 years ago
- Differentiable bitonic sorting☆141Updated 5 years ago
- Decoupled Neural Interfaces using Synthetic Gradients for PyTorch☆237Updated 6 years ago
- Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934☆113Updated 5 years ago
- Optimization with orthogonal constraints and on general manifolds☆129Updated 5 years ago
- [IJCAI'19, NeurIPS'19] Anode: Unconditionally Accurate Memory-Efficient Gradients for Neural ODEs☆106Updated 4 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆147Updated last year
- Code for experiments regarding importance sampling for training neural networks☆329Updated 3 years ago
- Original PyTorch implementation of the Leap meta-learner (https://arxiv.org/abs/1812.01054) along with code for running the Omniglot expe…☆148Updated 2 years ago
- TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".☆60Updated 6 years ago
- ICLR Reproducibility Challenge 2019☆218Updated 6 years ago
- This repository is no longer maintained. Check☆81Updated 5 years ago
- Lua implementation of Entropy-SGD☆82Updated 7 years ago
- Small Python library to automatically set CUDA_VISIBLE_DEVICES to the least loaded device on multi-GPU systems.☆107Updated 2 years ago