devzhk / cgds-package
Package for CGD and ACGD optimizers
☆19Updated 2 years ago
Related projects: ⓘ
- Refining continuous-in-depth neural networks☆39Updated 2 years ago
- repo for paper: Adaptive Checkpoint Adjoint (ACA) method for gradient estimation in neural ODE☆55Updated 3 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated last year
- Computing gradients and Hessians of feed-forward networks with GPU acceleration☆18Updated 7 months ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆34Updated 3 years ago
- Convex potential flows☆78Updated 2 years ago
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆66Updated 3 months ago
- Monotone operator equilibrium networks☆51Updated 4 years ago
- Machine Learning Function Approximation: This code implements the fully-connected Deep Neural Network (DNN) architectures considered in t…☆16Updated 3 years ago
- This repository contains code released by DiffEqML Research☆83Updated 2 years ago
- simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…☆14Updated 3 years ago
- Improved LBFGS and LBFGS-B optimizers in PyTorch.☆53Updated last week
- PyTorch implementation of the OT-Flow approach in arXiv:2006.00104☆49Updated last month
- ☆100Updated 3 years ago
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆35Updated last year
- Riemannian Convex Potential Maps☆68Updated last year
- Re-implementation of Hamiltonian Generative Networks paper☆32Updated 2 years ago
- ☆28Updated 5 years ago
- Model hub for all your DiffeqML needs. Pretrained weights, modules, and basic inference infrastructure☆20Updated last year
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆46Updated last year
- PyTorch implementation of GMLS-Nets. Machine learning methods for scattered unstructured data sets. Methods for learning differential op…☆22Updated 10 months ago
- Code for "Symplectic Adjoint Method for Exact Gradient of Neural ODE with Minimal Memory," NeurIPS, 2021.☆14Updated 2 years ago
- ☆78Updated 3 years ago
- ☆44Updated 10 months ago
- ☆33Updated 2 years ago
- Code for the ICLR 2020 paper "Learning to Control PDEs"☆29Updated 4 years ago
- A tool for generating PDEs ground truth datasets from ARCSim, FEniCS and SU2☆35Updated 3 years ago
- ☆28Updated last year
- Experiments for Meta-Learning Symmetries by Reparameterization☆55Updated 3 years ago
- Python and MATLAB code for Stein Variational sampling methods☆23Updated 5 years ago