#

kfac

Here are 5 public repositories matching this topic...

ikostrikov / pytorch-a2c-ppo-acktr-gail

PyTorch implementation of Advantage Actor Critic (A2C), Proximal Policy Optimization (PPO), Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation (ACKTR) and Generative Adversarial Imitation Learning (GAIL).

Updated May 29, 2022
Python

tfjgeorge / nngeometry

Manipulate {KFAC,EKFAC,Diagonal,Implicit} FIMs and finite width NTKs in PyTorch

pytorch fim ntk kfac fisher-information-matrix ekfac neural-tangent-kernel

Updated Mar 15, 2023
Python

alecwangcq / KFAC-Pytorch

Pytorch implementation of KFAC and E-KFAC (Natural Gradient).

optimization optimizer pytorch kfac ekfac

Updated Jul 2, 2019
Python

gpauloski / kfac-pytorch

Distributed K-FAC Preconditioner for PyTorch

deep-learning optimizer pytorch distributed kfac second-order-optimization

Updated Mar 21, 2023
Python

alirezakazemipour / ACKTR-PyTorch

a2c acktr kfac kfac-optimizer

Updated Aug 7, 2022
Python

Improve this page

Add a description, image, and links to the kfac topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the kfac topic, visit your repo's landing page and select "manage topics."