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☆132Updated last year
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆103Updated last year
- Differentiable interface to FEniCS for JAX☆58Updated 4 years ago
- Python Algorithms for Randomized Linear Algebra☆56Updated 2 years ago
- ☆73Updated 5 years ago
- Python trust-region subproblem solvers for nonlinear optimization☆29Updated 2 weeks ago
- Turning SymPy expressions into PyTorch modules.☆151Updated 2 years ago
- 18.336 - Fast Methods for Partial Differential and Integral Equations☆184Updated last year
- Exponential families for JAX☆75Updated this week
- ☆30Updated 3 years ago
- ☆112Updated 4 years ago
- ☆22Updated last year
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆58Updated 3 months ago
- Nonparametric Differential Equation Modeling☆55Updated last year
- Stencil computations in JAX☆71Updated 2 years ago
- Port-Hamiltonian Approach to Neural Network Training☆24Updated 5 years ago
- Minimal Implementation of Bayesian Optimization in JAX☆98Updated 6 months ago
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆65Updated 3 weeks ago
- Matrix-free linear algebra in JAX.☆145Updated last month
- Tools for JAX☆49Updated last week
- Repository for Deterministic Particle Flow Control framework☆10Updated 2 years ago
- Riemannian Optimization Using JAX☆53Updated 2 years ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆116Updated 3 weeks ago
- Code to estimate Renormalized Mutual Information in simple settings☆13Updated 4 years ago
- Minimal Gaussian process library in JAX with a simple (custom) approach to state management.☆12Updated last year
- ☆19Updated 3 years ago
- Tensor Train Toolbox☆123Updated this week
- Efficient Differentiable n-d PDE solvers in JAX.☆47Updated last week