gpauloski/kfac-pytorch

Distributed K-FAC Preconditioner for PyTorch

Language: Python - Size: 854 KB - Last synced: 2 days ago - Pushed: 3 days ago - Stars: 73 - Forks: 24

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).

Language: Python - Size: 5.91 MB - Last synced: 7 days ago - Pushed: about 2 years ago - Stars: 3,488 - Forks: 831

tfjgeorge/nngeometry

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

Language: Python - Size: 25.1 MB - Last synced: 9 days ago - Pushed: 24 days ago - Stars: 195 - Forks: 20

MaximeRobeyns/bayesian_lora

Bayesian Low-Rank Adaptation for Large Language Models

Language: Python - Size: 1.96 MB - Last synced: 14 days ago - Pushed: 14 days ago - Stars: 13 - Forks: 0

f-dangel/einconv

Convolutions and more as einsum for PyTorch

Language: Python - Size: 232 KB - Last synced: 8 days ago - Pushed: 6 months ago - Stars: 12 - Forks: 0

rafonsor/unRL

unRL (AKA "unreal") is a set of libraries providing Reinforcement Learning algorithms implemented in PyTorch or Jax.

Language: Python - Size: 354 KB - Last synced: 4 months ago - Pushed: 6 months ago - Stars: 0 - Forks: 0

alecwangcq/KFAC-Pytorch

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

Language: Python - Size: 29.3 KB - Last synced: 7 months ago - Pushed: almost 5 years ago - Stars: 106 - Forks: 32

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

Language: Python - Size: 40 KB - Last synced: 10 months ago - Pushed: 10 months ago - Stars: 1 - Forks: 0

alirezakazemipour/ACKTR-PyTorch

Language: Python - Size: 3.24 MB - Last synced: about 1 year ago - Pushed: almost 2 years ago - Stars: 0 - Forks: 0