deep-equilibrium-models topic

A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods

DiffEqML

High-performance and differentiation-enabled nonlinear solvers (Newton methods), bracketed rootfinding (bisection, Falsi), with sparsity and Newton-Krylov support.

SciML

Implicit Layer Machine Learning via Deep Equilibrium Networks, O(1) backpropagation with accelerated convergence.

SciML

PyTorch library for solving imaging inverse problems using deep learning

deepinv

Implementation of deep implicit attention in PyTorch

mcbal