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.☆58Updated 2 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆130Updated 11 months ago
- Differentiable interface to FEniCS for JAX☆57Updated 4 years ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆99Updated last year
- Turning SymPy expressions into PyTorch modules.☆150Updated 2 years ago
- Minimal Implementation of Bayesian Optimization in JAX☆95Updated 4 months ago
- 18.336 - Fast Methods for Partial Differential and Integral Equations☆185Updated last year
- Repository for Deterministic Particle Flow Control framework☆10Updated 2 years ago
- Python Algorithms for Randomized Linear Algebra☆56Updated 2 years ago
- ☆110Updated 4 years ago
- Python trust-region subproblem solvers for nonlinear optimization☆28Updated last year
- ☆19Updated 3 years ago
- Port-Hamiltonian Approach to Neural Network Training☆24Updated 5 years ago
- Stencil computations in JAX☆71Updated last year
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆112Updated 5 months ago
- Bayesian algorithm execution (BAX)☆50Updated 3 years ago
- ☆30Updated 3 years ago
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆26Updated 2 years ago
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆207Updated 3 weeks ago
- Code to estimate Renormalized Mutual Information in simple settings☆13Updated 4 years ago
- ☆33Updated 4 years ago
- Stiff Neural Ordinary Differential Equations☆35Updated 2 years ago
- Exponential families for JAX☆73Updated 3 weeks ago
- ☆21Updated 10 months ago
- ☆73Updated 4 years ago
- Scaling RLLib for generic simulation environments on Theta☆20Updated 2 years ago
- Solve sparse linear systems in JAX using the KLU algorithm☆36Updated 3 months ago
- Example code for paper: Automatic Differentiation to Simultaneously Identify Nonlinear Dynamics and Extract Noise Probability Distributio…☆56Updated 3 years ago
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆64Updated 2 months ago