tensorflow / kfac
An implementation of KFAC for TensorFlow
☆197Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for kfac
- ☆132Updated 7 years ago
- Decoupled Neural Interfaces using Synthetic Gradients for PyTorch☆236Updated 5 years ago
- Hypergradient descent☆138Updated 5 months ago
- ☆219Updated 6 years ago
- hessian in pytorch☆185Updated 4 years ago
- Forward-mode Automatic Differentiation for TensorFlow☆140Updated 6 years ago
- A colab that implements the Symplectic Gradient Adjustment optimizer from "The mechanics of n-player differentiable games"☆153Updated 5 years ago
- Implementation of Model-Agnostic Meta-Learning (MAML) in Jax☆188Updated 2 years ago
- Example code for Weight Normalization, from "Weight Normalization: A Simple Reparameterization to Accelerate Training of Deep Neural Netw…☆365Updated 5 years ago
- Lua implementation of Entropy-SGD☆81Updated 6 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆127Updated 5 years ago
- Optimizing control variates for black-box gradient estimation☆162Updated 5 years ago
- Guided Evolutionary Strategies☆264Updated last year
- Small Python library to automatically set CUDA_VISIBLE_DEVICES to the least loaded device on multi-GPU systems.☆106Updated 2 years ago
- Implementation of Spectral Inference Networks, ICLR 2019☆171Updated 5 years ago
- Hessian spectral density estimation in TF and Jax☆115Updated 4 years ago
- Sparse Variational Dropout, ICML 2017☆310Updated 4 years ago
- This repository is no longer maintained. Check☆82Updated 4 years ago
- Concise deep learning for JAX☆184Updated 4 years ago
- Second-order optimiser for deep networks☆76Updated 6 years ago
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆140Updated last year
- Code used to generate the results appearing in "Train longer, generalize better: closing the generalization gap in large batch training o…☆148Updated 7 years ago
- [IJCAI'19, NeurIPS'19] Anode: Unconditionally Accurate Memory-Efficient Gradients for Neural ODEs☆104Updated 4 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆141Updated last year
- Code for paper "L4: Practical loss-based stepsize adaptation for deep learning"☆124Updated 5 years ago
- Collection of algorithms for approximating Fisher Information Matrix for Natural Gradient (and second order method in general)☆135Updated 5 years ago
- Efficient PyTorch Hessian eigendecomposition tools!☆363Updated 8 months ago
- Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934☆111Updated 4 years ago
- ☆158Updated 3 months ago
- Neural Architecture Search with Bayesian Optimisation and Optimal Transport☆133Updated 5 years ago