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
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆131Updated 2 weeks ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆107Updated 2 years ago
- Differentiable interface to FEniCS for JAX☆58Updated 4 years ago
- Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.☆59Updated 3 years ago
- Turning SymPy expressions into PyTorch modules.☆156Updated 2 years ago
- Python trust-region subproblem solvers for nonlinear optimization☆30Updated last week
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆58Updated 5 months ago
- 18.336 - Fast Methods for Partial Differential and Integral Equations☆189Updated last year
- Python Algorithms for Randomized Linear Algebra☆57Updated 2 years ago
- Scaling RLLib for generic simulation environments on Theta☆20Updated 2 years ago
- ☆74Updated 5 years ago
- Stiff Neural Ordinary Differential Equations☆35Updated 2 years ago
- Repository for Deterministic Particle Flow Control framework☆10Updated 3 years ago
- ☆112Updated 4 years ago
- Efficient Differentiable n-d PDE solvers in JAX.☆52Updated 2 months ago
- Example code for paper: Automatic Differentiation to Simultaneously Identify Nonlinear Dynamics and Extract Noise Probability Distributio…☆60Updated 3 years ago
- Solve sparse linear systems in JAX using the KLU algorithm☆37Updated last week
- Code to accompany the paper "Discovery of Physics from Data: Universal Laws and Discrepancies"☆28Updated 5 years ago
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆69Updated last year
- SINDy (Sparse Identification of Nonlinear Dynamics) algorithms☆79Updated 3 years ago
- ☆28Updated 3 years ago
- Bayesian algorithm execution (BAX)☆55Updated 4 years ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆123Updated last month
- TorchFSM: Fourier Spectral Method with PyTorch☆53Updated 2 weeks ago
- ☆23Updated last year
- Stencil computations in JAX☆71Updated 2 months ago
- ☆30Updated 3 years ago
- Benchmark for learning stiff problems using physics-informed machine learning☆12Updated 4 years ago
- Port-Hamiltonian Approach to Neural Network Training☆26Updated 6 years ago