Preconditioned Conjugate Gradient in Pytorch
☆142Feb 14, 2024Updated 2 years ago
Alternatives and similar repositories for torch_cg
Users that are interested in torch_cg are comparing it to the libraries listed below
Sorting:
- Design preconditioners with a CNN to accelerate the conjugate gradient method.☆27Jul 2, 2025Updated 8 months ago
- Differentiation through cone programs☆117Jan 26, 2026Updated last month
- A PyTorch implementation of L-BFGS.☆625Feb 21, 2023Updated 3 years ago
- Anderson Acceleration☆20Feb 15, 2022Updated 4 years ago
- ☆14Jun 8, 2023Updated 2 years ago
- Interpolation routines in Pytorch.☆112Nov 30, 2020Updated 5 years ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆107Nov 12, 2023Updated 2 years ago
- Geometric Certifications of Neural Nets☆42Nov 22, 2022Updated 3 years ago
- Code Released for NeurIPS 2018 paper: Synthesized Policies for Transfer and Adaptation across Tasks and Environments☆16Apr 17, 2019Updated 6 years ago
- Slides from various talks I gave☆18Oct 25, 2018Updated 7 years ago
- A pytorch implementation of Amortized Stein Variational Gradient Descent/ Stein GAN☆18Dec 13, 2018Updated 7 years ago
- this is a case of seismic wave forward and inversion using PINN(physical-informed neural network)☆17Aug 25, 2022Updated 3 years ago
- A tensorflow implementation of the NIPS 2018 paper "Variational Inference with Tail-adaptive f-Divergence"☆20Jan 11, 2019Updated 7 years ago
- Collection of algorithms for approximating Fisher Information Matrix for Natural Gradient (and second order method in general)☆143May 26, 2019Updated 6 years ago
- Statistical adaptive stochastic optimization methods☆32Mar 28, 2020Updated 5 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆133Jul 2, 2019Updated 6 years ago
- Understanding Short-Horizon Bias in Stochastic Meta-Optimization☆37Mar 8, 2018Updated 7 years ago
- Tree-structured recurrent switching linear dynamical systems☆38Jul 13, 2020Updated 5 years ago
- Efficient PyTorch Hessian eigendecomposition tools!☆386Feb 29, 2024Updated 2 years ago
- ☆10Jul 6, 2022Updated 3 years ago
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆153Jun 22, 2023Updated 2 years ago
- Pytorch implementation of Recurrent Neural Processes https://arxiv.org/pdf/1906.05915.pdf☆22Aug 6, 2019Updated 6 years ago
- An extensible benchmark suite to evaluate data-driven physical simulation☆32Dec 8, 2021Updated 4 years ago
- Automatic Integration for Neural Spatio-Temporal Point Process models (AI-STPP) is a new paradigm for exact, efficient, non-parametric inf…☆25Oct 14, 2024Updated last year
- ☆37Feb 17, 2022Updated 4 years ago
- Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561☆25Mar 30, 2021Updated 4 years ago
- Supporting code for "Parallel Streaming Wasserstein Barycenters"☆10Nov 14, 2017Updated 8 years ago
- Cone program refinement☆10Mar 6, 2020Updated 6 years ago
- Implementation of Receding Horizon Curiosity Algrithm☆13Mar 24, 2023Updated 2 years ago
- Frontier GPU multi-physics solvers☆13Feb 16, 2026Updated 2 weeks ago
- Tools for generating cubed sphere grids and solving partial differential equations on the sphere.☆15Jan 20, 2026Updated last month
- code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"☆26Dec 10, 2022Updated 3 years ago
- A second-order optimizer for deep networks☆25Oct 31, 2019Updated 6 years ago
- PyTorch Implementation of Unrolled Generative Adversarial Networks☆40Jun 27, 2018Updated 7 years ago
- The original code for the paper "How to train your MAML" along with a replication of the original "Model Agnostic Meta Learning" (MAML) p…☆41Oct 28, 2020Updated 5 years ago
- A working implementation of the Categorical DQN (Distributional RL).☆95Apr 7, 2018Updated 7 years ago
- Pytorch implementation of 'Semi-Implicit Methods for Deep Neural Networks'☆25May 13, 2019Updated 6 years ago
- Proximal Backpropagation - a neural network training algorithm that takes implicit instead of explicit gradient steps☆42Mar 17, 2019Updated 6 years ago
- Implementation of radon transform for pytorch☆12Aug 31, 2020Updated 5 years ago