Bringing PDEs to JAX with forward and reverse modes automatic differentiation

31 August 2023

Papers citing "Bringing PDEs to JAX with forward and reverse modes automatic differentiation"

5 / 5 papers shown

Title
Physics-driven machine learning models coupling PyTorch and Firedrake N. Bouziani David A. Ham AI4CE 14 3 0 13 Mar 2023
j-Wave: An open-source differentiable wave simulator A. Stanziola Simon Arridge B. Cox B. Treeby VLM 33 21 0 30 Jun 2022
Randomized Maximum Likelihood via High-Dimensional Bayesian Optimization Valentin Breaz Richard D. Wilkinson 19 0 0 17 Apr 2022
Adjoint-aided inference of Gaussian process driven differential equations Paterne Gahungu Christopher W. Lanyon Mauricio A. Alvarez Engineer Bainomugisha M. Smith Richard D. Wilkinson 16 5 0 09 Feb 2022
A research framework for writing differentiable PDE discretizations in JAX A. Stanziola Simon Arridge B. Cox B. Treeby 21 8 0 09 Nov 2021