mazy1998 / Solving-Optimization-Problems-with-JAXLinks
Solving Optimization Problems with JAX, code and PDF
☆16Updated 5 years ago
Alternatives and similar repositories for Solving-Optimization-Problems-with-JAX
Users that are interested in Solving-Optimization-Problems-with-JAX are comparing it to the libraries listed below
Sorting:
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.☆59Updated 3 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆131Updated last year
- Differentiable interface to FEniCS for JAX☆58Updated 4 years ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆105Updated 2 years ago
- Turning SymPy expressions into PyTorch modules.☆153Updated 2 years ago
- 18.336 - Fast Methods for Partial Differential and Integral Equations☆186Updated last year
- Stencil computations in JAX☆71Updated 3 weeks ago
- ☆30Updated 3 years ago
- Python Algorithms for Randomized Linear Algebra☆57Updated 2 years ago
- ☆22Updated last year
- Python trust-region subproblem solvers for nonlinear optimization☆29Updated last week
- Repository for Deterministic Particle Flow Control framework☆10Updated 3 years ago
- Minimal Implementation of Bayesian Optimization in JAX☆100Updated 7 months ago
- Exponential families for JAX☆75Updated last month
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆58Updated 4 months ago
- ☆112Updated 4 years ago
- Bayesian algorithm execution (BAX)☆55Updated 4 years ago
- Normalizing Flows using JAX☆85Updated 2 years ago
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆26Updated 2 years ago
- ☆28Updated 4 years ago
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆211Updated 4 months ago
- Stiff Neural Ordinary Differential Equations☆35Updated 2 years ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 4 years ago
- Port-Hamiltonian Approach to Neural Network Training☆24Updated 5 years ago
- ☆28Updated 3 years ago
- ☆73Updated 5 years ago
- Code to estimate Renormalized Mutual Information in simple settings☆13Updated 4 years ago
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆69Updated last year
- Nonparametric Differential Equation Modeling☆56Updated last year