pnkraemer / million-dimension-prob-ode-solver-experiments
☆10Updated 2 years ago
Alternatives and similar repositories for million-dimension-prob-ode-solver-experiments:
Users that are interested in million-dimension-prob-ode-solver-experiments are comparing it to the libraries listed below
- Probabilistic Finite Volume Method based on Affine Gaussian Process inference☆10Updated 7 months ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆38Updated 3 months ago
- an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets☆22Updated last year
- Efficient Differentiable n-d PDE solvers in JAX.☆23Updated 2 months ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆42Updated last year
- Convolutional Differential Operators for Physics-based Deep Learning Study☆24Updated 6 months ago
- Website for the book "The Elements of Differentiable Programming".☆13Updated 4 months ago
- Differentiable interface to FEniCS for JAX☆52Updated 3 years ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆49Updated last year
- Official PyTorch implementation of the Vectorized Conditional Neural Field.☆12Updated 5 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆124Updated 4 months ago
- Tutorial materials of the Probabilistic Numerics Spring School.☆34Updated last year
- Matrix-free linear algebra in JAX.☆113Updated 3 weeks ago
- Code for the book "The Elements of Differentiable Programming".☆71Updated 5 months ago
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆35Updated 2 years ago
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆58Updated last week
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆20Updated 6 months ago
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Updated last year
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆25Updated 9 months ago
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- [ICML 2022] Learning Efficient and Robust Ordinary Differential \\ Equations via Invertible Neural Networks☆9Updated last year
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆92Updated last year
- Library for normalizing flows and neural flows.☆24Updated 2 years ago
- An unofficial implementation of the Fourier Neural Operator in Flax☆17Updated 8 months ago
- ☆20Updated 3 months ago
- Code for the paper "Rational neural networks", NeurIPS 2020☆28Updated 3 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- Repository for Deterministic Particle Flow Control framework☆10Updated 2 years ago
- Distributed Fourier Neural Operators☆29Updated 2 years ago
- Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)☆18Updated 3 years ago