n-gao / pytorch-kfacView external linksLinks
Pytorch implementation of KFAC - this is a port of https://github.com/tensorflow/kfac/
☆30Jun 6, 2024Updated last year
Alternatives and similar repositories for pytorch-kfac
Users that are interested in pytorch-kfac are comparing it to the libraries listed below
Sorting:
- SKFAC Preconditioner for MindSpore☆12Jul 2, 2021Updated 4 years ago
- Finite-temperature variational Monte Carlo calculation of uniform electron gas using neural canonical transformation.☆16Jun 17, 2023Updated 2 years ago
- Reference implementation of "Ab-Initio Potential Energy Surfaces by Pairing GNNs with Neural Wave Functions" (ICLR, 2022) and "Sampling-f…☆32Jun 14, 2024Updated last year
- ☆15Apr 19, 2020Updated 5 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆132Jul 2, 2019Updated 6 years ago
- Repo for "PIX2NVS: Parameterized conversion of pixel-domain video frames to neuromorphic vision streams" ICIP2017☆22Oct 4, 2021Updated 4 years ago
- Experiments on the impact of depth in transformers and SSMs.☆40Oct 23, 2025Updated 3 months ago
- Implementation of Forward Laplacian algorithm in JAX☆94Nov 25, 2025Updated 2 months ago
- A Universal Neural Network Solver for The Schrödinger Equation☆10Dec 3, 2024Updated last year
- This repo contains all the code related to the interactive online Bloch sphere simulator☆18Jan 19, 2026Updated 3 weeks ago
- Implementation of Pre-text invariant representation learning algorithm in pytorch☆11May 27, 2020Updated 5 years ago
- A data processing module implemented with numpy☆10Aug 16, 2022Updated 3 years ago
- Towards Understanding Sharpness-Aware Minimization [ICML 2022]☆38Jun 14, 2022Updated 3 years ago
- Implementations of the renormalization group-based diffusion model (RGDM).☆16Mar 10, 2025Updated 11 months ago
- Demo Library for Numato FPGA Boards☆12Mar 4, 2015Updated 10 years ago
- A Transfer Matrix Method Toolkit in Matlab☆12Oct 9, 2024Updated last year
- Code for Lagrangian Hashes for Compressed Neural Fields Representation☆12Sep 24, 2024Updated last year
- Generalized Method of Moments estimation☆13Mar 23, 2025Updated 10 months ago
- Code for the NeurIPS 2024 submission: "DAGER: Extracting Text from Gradients with Language Model Priors"☆20Aug 13, 2025Updated 6 months ago
- Bidirectional transformation between Yao IR and QASM.☆11Dec 6, 2020Updated 5 years ago
- ☆10Jun 13, 2021Updated 4 years ago
- Some commonly used functions and modules☆10Jan 15, 2024Updated 2 years ago
- Agent models implemented with Pyro☆11Jul 11, 2023Updated 2 years ago
- ☆10Dec 5, 2020Updated 5 years ago
- Demos for the 2022 Many Electron Collaboration Workshop on PySCF☆12Jun 21, 2022Updated 3 years ago
- ☆14Jul 17, 2024Updated last year
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆11May 8, 2023Updated 2 years ago
- Proxy for OpenAI☆15Sep 2, 2025Updated 5 months ago
- A PyTorch implementation of SimSiam based on CVPR 2021 paper "Exploring Simple Siamese Representation Learning"☆12Mar 23, 2021Updated 4 years ago
- mPLM-Sim: Better Cross-Lingual Similarity and Transfer in Multilingual Pretrained Language Models☆11Jan 19, 2024Updated 2 years ago
- Code for Q-learning with parametrized quantum circuits in OpenAI Gym environments.☆12Nov 12, 2021Updated 4 years ago
- A PyTorch implementation of WRN☆10Jun 2, 2018Updated 7 years ago
- basic install & update shell scripts for Ubuntu☆10Sep 2, 2020Updated 5 years ago
- The implementation of the Block Coordinate Regularization by Denoising (BC-RED) algorithm (NeurIPS 2019)☆10Oct 15, 2019Updated 6 years ago
- ☆11Jan 23, 2017Updated 9 years ago
- ☆13Jun 2, 2022Updated 3 years ago
- [ICCV 2025] Official repository of the paper "BlinkTrack: Feature Tracking over 80 FPS via Events and Images". This repository contains t…☆11Aug 27, 2025Updated 5 months ago
- Modification of SOMPY repo with robust K-means clustering (bootstrapped SSE elbow method)☆13Apr 6, 2019Updated 6 years ago
- Additional features supporting paclets in Mathematica☆15Oct 3, 2017Updated 8 years ago