tfjgeorge/nngeometry

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

Language: Python - Size: 25.1 MB - Last synced at: 5 days ago - Pushed at: 5 days ago - Stars: 211 - Forks: 23

gpauloski/kfac-pytorch

Distributed K-FAC Preconditioner for PyTorch

Language: Python - Size: 922 KB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 85 - Forks: 25

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 at: 8 days ago - Pushed at: almost 3 years ago - Stars: 3,729 - Forks: 835

f-dangel/einconv

Convolutions and more as einsum for PyTorch

Language: Python - Size: 250 KB - Last synced at: 20 days ago - Pushed at: 11 months ago - Stars: 15 - Forks: 1

MaximeRobeyns/bayesian_lora

Bayesian Low-Rank Adaptation for Large Language Models

Language: Python - Size: 1.74 MB - Last synced at: 9 days ago - Pushed at: 10 months ago - Stars: 30 - Forks: 5

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 at: 19 days ago - Pushed at: over 1 year ago - Stars: 4 - Forks: 1

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 at: about 1 year ago - Pushed at: over 1 year 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 at: over 1 year ago - Pushed at: almost 6 years ago - Stars: 106 - Forks: 32

alirezakazemipour/ACKTR-PyTorch

Language: Python - Size: 3.24 MB - Last synced at: about 2 years ago - Pushed at: over 2 years ago - Stars: 0 - Forks: 0