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
- Hypergradient descent☆149Updated last year
- Neural Architecture Search with Bayesian Optimisation and Optimal Transport☆135Updated 6 years ago
- Code for NeurIPS 2019 paper: "Tensor Programs I: Wide Feedforward or Recurrent Neural Networks of Any Architecture are Gaussian Processes…☆246Updated 5 years ago
- Optimizing control variates for black-box gradient estimation☆163Updated 6 years ago
- A colab that implements the Symplectic Gradient Adjustment optimizer from "The mechanics of n-player differentiable games"☆153Updated 6 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆147Updated last year
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆146Updated 2 years ago
- Optimization with orthogonal constraints and on general manifolds☆130Updated 5 years ago
- Implementation of Spectral Inference Networks, ICLR 2019☆172Updated 6 years ago
- Tensor Train decomposition on TensorFlow☆225Updated 4 years ago
- Regularization, Neural Network Training Dynamics☆14Updated 5 years ago
- ☆219Updated 7 years ago
- Experiments for the paper "Exponential expressivity in deep neural networks through transient chaos"☆72Updated 9 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆132Updated 6 years ago
- Convolutional Neural Tangent Kernel☆113Updated 5 years ago
- This repository is no longer maintained. Check☆81Updated 5 years ago
- ☆170Updated last year
- Hessian spectral density estimation in TF and Jax☆124Updated 5 years ago
- Lua implementation of Entropy-SGD☆82Updated 7 years ago
- Decoupled Neural Interfaces using Synthetic Gradients for PyTorch☆238Updated 6 years ago
- Guided Evolutionary Strategies☆272Updated 2 years ago
- Forward-mode Automatic Differentiation for TensorFlow☆139Updated 7 years ago
- Concise deep learning for JAX☆183Updated 4 years ago
- Autoregressive Energy Machines☆78Updated 2 years ago
- Sparse Variational Dropout, ICML 2017☆313Updated 5 years ago
- Small Python library to automatically set CUDA_VISIBLE_DEVICES to the least loaded device on multi-GPU systems.☆108Updated 2 years ago
- Code for experiments regarding importance sampling for training neural networks☆329Updated 3 years ago
- Implementation of Model-Agnostic Meta-Learning (MAML) in Jax☆191Updated 2 years ago