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
- Differentiable interface to FEniCS for JAX☆54Updated 4 years ago
- Exponential families for JAX☆69Updated this week
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 4 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆128Updated 8 months ago
- Efficient Differentiable n-d PDE solvers in JAX.☆39Updated 7 months ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆96Updated last year
- ☆29Updated 2 years ago
- Solve sparse linear systems in JAX using the KLU algorithm☆35Updated 3 weeks ago
- ☆72Updated 4 years ago
- Repository for Deterministic Particle Flow Control framework☆10Updated 2 years ago
- ☆21Updated 7 months ago
- PINNs-JAX, Physics-informed Neural Networks (PINNs) implemented in JAX.☆48Updated 8 months ago
- Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.☆58Updated 2 years ago
- Tools for JAX☆47Updated last week
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- Benchmark for learning stiff problems using physics-informed machine learning☆12Updated 3 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆53Updated last month
- Stencil computations in JAX☆71Updated last year
- Riemannian Optimization Using JAX☆49Updated last year
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆69Updated last year
- ☆19Updated 2 years ago
- A PyTorch library for all things nonlinear control and reinforcement learning.☆46Updated 3 years ago
- Port-Hamiltonian Approach to Neural Network Training☆24Updated 5 years ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆43Updated last year
- A Python library for training neural ODEs.☆21Updated 3 months ago
- Stiff Neural Ordinary Differential Equations☆34Updated 2 years ago
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆57Updated 6 months ago
- Differentiable interface to Firedrake for JAX☆14Updated 4 years ago
- Bayesian inference with Python and Jax.☆32Updated 2 years ago