mazy1998 / Solving-Optimization-Problems-with-JAXLinks
Solving Optimization Problems with JAX, code and PDF
☆17Updated 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☆132Updated 2 weeks ago
- Differentiable interface to FEniCS for JAX☆58Updated 4 years ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆107Updated 2 years ago
- Port-Hamiltonian Approach to Neural Network Training☆27Updated 6 years ago
- Turning SymPy expressions into PyTorch modules.☆156Updated 2 years ago
- Python trust-region subproblem solvers for nonlinear optimization☆30Updated 3 weeks ago
- Python Algorithms for Randomized Linear Algebra☆57Updated 2 years ago
- Matrix-free linear algebra in JAX.☆159Updated 2 months ago
- 18.336 - Fast Methods for Partial Differential and Integral Equations☆190Updated last week
- ☆74Updated 5 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆58Updated 6 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆123Updated this week
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆53Updated 2 months ago
- ☆112Updated 4 years ago
- Exponential families for JAX☆76Updated last week
- Minimal Implementation of Bayesian Optimization in JAX☆102Updated 9 months ago
- Stencil computations in JAX☆71Updated 2 months ago
- Solve sparse linear systems in JAX using the KLU algorithm☆38Updated last week
- Repository for Deterministic Particle Flow Control framework☆10Updated 3 years ago
- ☆23Updated last year
- Code to estimate Renormalized Mutual Information in simple settings☆13Updated 4 years ago
- Riemannian Optimization Using JAX☆53Updated 2 years ago
- ☆30Updated 3 years ago
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆68Updated 2 months ago
- Normalizing Flows using JAX☆86Updated 2 years ago
- Bayesian algorithm execution (BAX)☆55Updated 4 years ago
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆214Updated 6 months ago
- Nonparametric Differential Equation Modeling☆56Updated last year