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
- Efficient Differentiable n-d PDE solvers in JAX.☆24Updated 3 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆123Updated 4 months ago
- 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
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆49Updated last year
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆41Updated 3 months ago
- Probabilistic Finite Volume Method based on Affine Gaussian Process inference☆10Updated 8 months ago
- Differentiable interface to FEniCS for JAX☆52Updated 3 years ago
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆25Updated 9 months ago
- An unofficial implementation of the Fourier Neural Operator in Flax☆17Updated 8 months ago
- Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)☆18Updated 3 years ago
- Convolutional Differential Operators for Physics-based Deep Learning Study☆24Updated 6 months ago
- Neural Emulator Architectures in JAX.☆16Updated 3 months ago
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆70Updated 8 months ago
- Tutorial materials of the Probabilistic Numerics Spring School.☆34Updated last year
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆82Updated this week
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆59Updated 3 months ago
- Website for the book "The Elements of Differentiable Programming".☆13Updated 5 months ago
- Spectral Neural Operator☆75Updated last year
- Repository for Deterministic Particle Flow Control framework☆10Updated 2 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆43Updated last month
- ☆31Updated last year
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆93Updated last year
- Code for the book "The Elements of Differentiable Programming".☆72Updated this week
- an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets☆22Updated last year
- Practicum on Supervised Learning in Function Spaces☆32Updated 2 years ago
- Distributed Fourier Neural Operators☆29Updated 2 years ago
- Riemannian Optimization Using JAX☆48Updated last year
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆58Updated 3 weeks ago
- Matrix-free linear algebra in JAX.☆114Updated last month
- Stencil computations in JAX☆70Updated last year