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
- hessian in pytorch☆187Updated 4 years ago
- Optimizing control variates for black-box gradient estimation☆163Updated 5 years ago
- ☆219Updated 7 years ago
- Decoupled Neural Interfaces using Synthetic Gradients for PyTorch☆237Updated 6 years ago
- Forward-mode Automatic Differentiation for TensorFlow☆139Updated 7 years ago
- Implementation of Model-Agnostic Meta-Learning (MAML) in Jax☆189Updated 2 years ago
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆143Updated last year
- A colab that implements the Symplectic Gradient Adjustment optimizer from "The mechanics of n-player differentiable games"☆154Updated 6 years ago
- Small Python library to automatically set CUDA_VISIBLE_DEVICES to the least loaded device on multi-GPU systems.☆107Updated 2 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆132Updated 5 years ago
- Hessian spectral density estimation in TF and Jax☆123Updated 4 years ago
- Concise deep learning for JAX☆183Updated 4 years ago
- Second-order optimiser for deep networks☆76Updated 6 years ago
- Hypergradient descent☆148Updated last year
- [IJCAI'19, NeurIPS'19] Anode: Unconditionally Accurate Memory-Efficient Gradients for Neural ODEs☆104Updated 4 years ago
- This repository is no longer maintained. Check☆81Updated 5 years ago
- Code used to generate the results appearing in "Train longer, generalize better: closing the generalization gap in large batch training o…☆149Updated 8 years ago
- ☆168Updated 10 months ago
- Guided Evolutionary Strategies☆271Updated 2 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆146Updated last year
- Implementation of Spectral Inference Networks, ICLR 2019☆172Updated 6 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
- Code for paper "L4: Practical loss-based stepsize adaptation for deep learning"☆124Updated 6 years ago
- Collection of algorithms for approximating Fisher Information Matrix for Natural Gradient (and second order method in general)☆139Updated 6 years ago
- Optimization with orthogonal constraints and on general manifolds☆128Updated 4 years ago
- ☆157Updated last year
- Autoregressive Energy Machines☆78Updated 2 years ago
- Regularization, Neural Network Training Dynamics☆14Updated 5 years ago
- A Python implementation of the gradient REBAR estimator.☆46Updated 6 years ago