Pytorch implementation of KFAC - this is a port of https://github.com/tensorflow/kfac/
☆33Jun 6, 2024Updated 2 years ago
Alternatives and similar repositories for pytorch-kfac
Users that are interested in pytorch-kfac are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- SKFAC Preconditioner for MindSpore☆12Jul 2, 2021Updated 5 years ago
- Distributed K-FAC preconditioner for PyTorch☆98Updated this week
- simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…☆17Oct 21, 2020Updated 6 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆135Jul 2, 2019Updated 7 years ago
- Reference implementation of "Ab-Initio Potential Energy Surfaces by Pairing GNNs with Neural Wave Functions" (ICLR, 2022) and "Sampling-f…☆34Jun 14, 2024Updated 2 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- ☆137Oct 23, 2017Updated 9 years ago
- Implementation of Forward Laplacian algorithm in JAX☆107Updated this week
- Finite-temperature variational Monte Carlo calculation of uniform electron gas using neural canonical transformation.☆16Jun 17, 2023Updated 3 years ago
- An implementation of KFAC for TensorFlow☆198Feb 11, 2022Updated 4 years ago
- Node Equivariant Message Passing Neural Network☆12Jul 4, 2026Updated 3 months ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆332Oct 13, 2026Updated 2 weeks ago
- ☆25Mar 1, 2022Updated 4 years ago
- A Universal Neural Network Solver for The Schrödinger Equation☆12Oct 3, 2026Updated 3 weeks ago
- A physics-informed neural network method for head-related transfer function upsampling☆10Feb 27, 2025Updated last year
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- A Transfer Matrix Method Toolkit in Matlab☆11Oct 9, 2024Updated 2 years ago
- Differentiable N-body code (JAX)☆11May 8, 2025Updated last year
- DeepErwin is a python 3.8+ package that implements and optimizes JAX 2.x wave function models for numerical solutions to the multi-electr…☆67Apr 18, 2025Updated last year
- A simple Jax implementation of influence functions.☆21Apr 9, 2024Updated 2 years ago
- ESI-DCAFM-TACO-VDSP Summer School on "Machine Learning for Materials Hard and Soft"☆40Jul 22, 2022Updated 4 years ago
- ☆13Aug 15, 2024Updated 2 years ago
- ☆17Jun 16, 2026Updated 4 months ago
- Piano Chords☆16May 9, 2026Updated 5 months ago
- Bidirectional transformation between Yao IR and QASM.☆11Dec 6, 2020Updated 5 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- For advanced physics-driven combined with neural network enhancement force field.☆19Mar 9, 2026Updated 7 months ago
- A framework based on Tensorflow for running variational Monte-Carlo simulations of quantum many-body systems.☆37Oct 30, 2020Updated 6 years ago
- Deep Reinforcement Learning Toolbox for Robotics using Keras and TensorFlow☆19Aug 16, 2019Updated 7 years ago
- ☆13Jul 2, 2021Updated 5 years ago
- A toolbox for inference of mixture models☆16May 29, 2023Updated 3 years ago
- A library for symmetry operations and linear interpolation within an irreducible part of the first Brillouin zone.☆17Oct 8, 2026Updated 3 weeks ago
- ☆22Nov 29, 2021Updated 4 years ago
- ☆11Dec 15, 2024Updated last year
- [ICLR 2023] Eva: Practical Second-order Optimization with Kronecker-vectorized Approximation☆12Jul 31, 2023Updated 3 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Optimizing the Optimizer for Physics-Informed Neural Networks and Kolmogorov-Arnold Networks☆16May 29, 2026Updated 5 months ago
- Towards Understanding Sharpness-Aware Minimization [ICML 2022]☆38Jun 14, 2022Updated 4 years ago
- Python 3.6 and TensorFlow implementation of the AReS and MaRS algorithms☆11Jun 23, 2019Updated 7 years ago
- Code for Knowledge-Adaptation Priors based on the NeurIPS 2021 paper by Khan and Swaroop.☆16Feb 1, 2022Updated 4 years ago
- A package for creating a matrix product state (MPS) from a free fermion (Gaussian) state.☆10Dec 6, 2021Updated 4 years ago
- ☆13Feb 12, 2023Updated 3 years ago
- Stochastic Series Expansion☆11Jun 5, 2023Updated 3 years ago