facebookresearch / neural-scs
Neural Fixed-Point Acceleration for Convex Optimization
☆29Updated 2 years ago
Alternatives and similar repositories for neural-scs:
Users that are interested in neural-scs are comparing it to the libraries listed below
- Variational Reinforcement Learning☆16Updated 6 months ago
- ☆49Updated 4 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- ☆13Updated 5 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- Code for Unbiased Implicit Variational Inference (UIVI)☆13Updated 6 years ago
- Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"☆15Updated 2 years ago
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆43Updated 3 years ago
- Source code for the Paper: CombOptNet: Fit the Right NP-Hard Problem by Learning Integer Programming Constraints}☆72Updated 2 years ago
- Source code for paper Conservative Uncertainty Estimation By Fitting Prior Networks (ICLR 2020)☆21Updated 2 years ago
- Monotone operator equilibrium networks☆51Updated 4 years ago
- ☆28Updated 5 years ago
- ☆18Updated 3 years ago
- The differentiable OSQP solver layer for PyTorch☆59Updated 2 years ago
- ☆31Updated 4 years ago
- Inference on non-linear dynamical systems written in JAX☆11Updated 4 years ago
- repo for paper: Adaptive Checkpoint Adjoint (ACA) method for gradient estimation in neural ODE☆54Updated 3 years ago
- Riemannian Convex Potential Maps☆67Updated last year
- Contact-Aware Symplectic Integrator Network☆11Updated last year
- [ICLR 2022] Path integral sampler☆43Updated last year
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆27Updated 3 years ago
- Public Release of Plan2vec Implementation in pyTorch☆56Updated 2 years ago
- Code accompanying VarGrad: A Low-Variance Gradient Estimator for Variational Inference☆12Updated 4 years ago
- Variational Auto-Regressive Gaussian Processes for Continual Learning☆20Updated 3 years ago
- Euclidean Wasserstein-2 optimal transportation☆44Updated last year
- Source code of "Grid-to-Graph: Flexible Spatial Relational Inductive Biases for Reinforcement Learning" (AAMAS 2021).☆27Updated 3 years ago
- Code to minimize the Variational Contrastive Divergence (VCD)☆29Updated 5 years ago
- Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561☆24Updated 3 years ago
- Generalised UDRL☆37Updated 2 years ago
- Hessian trace estimation using PyTorch and Hutch++☆19Updated 4 years ago