kfac topic

List kfac repositories

pytorch-a2c-ppo-acktr-gail

3.5k
Stars
831
Forks
Watchers

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

nngeometry

215
Stars
24
Forks
215
Watchers

{KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch

KFAC-Pytorch

120
Stars
33
Forks
Watchers

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

kfac-pytorch

73
Stars
24
Forks
Watchers

Distributed K-FAC Preconditioner for PyTorch

bayesian_lora

15
Stars
0
Forks
Watchers

Bayesian Low-Rank Adaptation for Large Language Models