kfac
There are 9 repositories under kfac 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).
tfjgeorge/nngeometry
{KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch
alecwangcq/KFAC-Pytorch
Pytorch implementation of KFAC and E-KFAC (Natural Gradient).
gpauloski/kfac-pytorch
Distributed K-FAC Preconditioner for PyTorch
MaximeRobeyns/bayesian_lora
Bayesian Low-Rank Adaptation for Large Language Models
f-dangel/einconv
Convolutions and more as einsum for PyTorch
yorkerlin/StructuredNGD-DL
Matrix-multiplication-only KFAC; Code for ICML 2023 paper on Simplifying Momentum-based Positive-definite Submanifold Optimization with Applications to Deep Learning
rafonsor/unRL
unRL (AKA "unreal") is a set of libraries providing Reinforcement Learning algorithms implemented in PyTorch or Jax.