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
- ☆219Updated 7 years ago
- A colab that implements the Symplectic Gradient Adjustment optimizer from "The mechanics of n-player differentiable games"☆154Updated 6 years ago
- Forward-mode Automatic Differentiation for TensorFlow☆139Updated 7 years ago
- hessian in pytorch☆187Updated 4 years ago
- Guided Evolutionary Strategies☆271Updated 2 years ago
- Optimizing control variates for black-box gradient estimation☆163Updated 5 years ago
- Differentiable bitonic sorting☆141Updated 4 years ago
- This repository is no longer maintained. Check☆81Updated 5 years ago
- Decoupled Neural Interfaces using Synthetic Gradients for PyTorch☆237Updated 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
- ☆168Updated 10 months ago
- Implementation of Spectral Inference Networks, ICLR 2019☆172Updated 6 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆132Updated 5 years ago
- [IJCAI'19, NeurIPS'19] Anode: Unconditionally Accurate Memory-Efficient Gradients for Neural ODEs☆104Updated 4 years ago
- Experiments for the paper "Exponential expressivity in deep neural networks through transient chaos"☆71Updated 9 years ago
- Small Python library to automatically set CUDA_VISIBLE_DEVICES to the least loaded device on multi-GPU systems.☆107Updated 2 years ago
- Hypergradient descent☆148Updated last year
- Second-order optimiser for deep networks☆76Updated 6 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆146Updated last year
- Lua implementation of Entropy-SGD☆82Updated 7 years ago
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆143Updated 2 years ago
- Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934☆113Updated 5 years ago
- DeepOBS: A Deep Learning Optimizer Benchmark Suite☆106Updated last year
- Convolutional Neural Tangent Kernel☆111Updated 5 years ago
- Concise deep learning for JAX☆183Updated 4 years ago
- Code for "A-NICE-MC: Adversarial Training for MCMC"☆125Updated 6 years ago
- Collection of algorithms for approximating Fisher Information Matrix for Natural Gradient (and second order method in general)☆139Updated 6 years ago
- Hessian spectral density estimation in TF and Jax☆123Updated 4 years ago
- Optimization with orthogonal constraints and on general manifolds☆128Updated 4 years ago