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
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆128Updated 9 months ago
- Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.☆58Updated 2 years ago
- Differentiable interface to FEniCS for JAX☆54Updated 4 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆54Updated last month
- ☆30Updated 2 years ago
- Efficient Differentiable n-d PDE solvers in JAX.☆40Updated 7 months ago
- Code to accompany the paper "Discovery of Physics from Data: Universal Laws and Discrepancies"☆27Updated 5 years ago
- Repository for Deterministic Particle Flow Control framework☆10Updated 2 years ago
- Riemannian Optimization Using JAX☆49Updated last year
- Python trust-region subproblem solvers for nonlinear optimization☆27Updated 11 months ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆43Updated last year
- Code to estimate Renormalized Mutual Information in simple settings☆13Updated 4 years ago
- By introducing a differentiable contact model, DiffCoSim extends the applicability of Lagrangian/Hamiltonian-inspired neural networks to …☆36Updated 2 years ago
- ☆21Updated 8 months ago
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- [ICML 2022] Learning Efficient and Robust Ordinary Differential \\ Equations via Invertible Neural Networks☆10Updated 2 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 5 months ago
- We simulate a wind tunnel, place a rectangular occlusion in it, and then use gradient descent to turn the occlusion into a wing.☆26Updated 4 years ago
- Port-Hamiltonian Approach to Neural Network Training☆24Updated 5 years ago
- ☆20Updated 2 years ago
- Tools for JAX☆47Updated last week
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆40Updated 4 years ago
- Material for 'Mathematics of Deep Learning Workshop' (Invited Talk)☆17Updated 2 years ago
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆24Updated 2 years ago
- A collection of graph neural networks implementations in JAX☆33Updated last year
- Code for the paper "Variational Monte Carlo Approach to Partial Differential Equations with Neural Networks" (https://arxiv.org/abs/2206.…☆10Updated 2 years ago
- A differentiation API for PyTorch☆30Updated 5 years ago
- Differentiable interface to Firedrake for JAX☆14Updated 4 years ago
- ☆29Updated 2 years ago