Saved in:
Bibliographic Details
Main Authors: Hipperson, James, Hargreaves, Jonathan, Cox, Trevor
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2604.21431
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917431077765120
author Hipperson, James
Hargreaves, Jonathan
Cox, Trevor
author_facet Hipperson, James
Hargreaves, Jonathan
Cox, Trevor
contents Engineering structures are increasingly designed using numerical optimisation. However, traditional optimisation methods can be challenging with multiple objectives and many parameters. In machine learning, stable training of artificial neural networks with millions or billions of parameters is achieved using automatic differentiation frameworks such as JAX and Pytorch. Because these frameworks provide accelerated numerical linear algebra with automatic gradient tracking, they also enable differentiable implementations of numerical methods to be built. This facilitates faster gradient-based optimisation of geometry and materials, as well as solution of inverse problems. We demonstrate JAX-BEM, a differentiable Boundary Element Method (BEM) solver, showing that it matches the error of existing BEM codes for a benchmark problem and enables gradient-based geometry optimisation. Although the demonstrated examples are for acoustic simulations, the concept could be readily extended to electromagnetic waves.
format Preprint
id arxiv_https___arxiv_org_abs_2604_21431
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle JAX-BEM: Gradient-Based Acoustic Shape Optimisation via a Differentiable Boundary Element Method
Hipperson, James
Hargreaves, Jonathan
Cox, Trevor
Computational Engineering, Finance, and Science
Computational Physics
Engineering structures are increasingly designed using numerical optimisation. However, traditional optimisation methods can be challenging with multiple objectives and many parameters. In machine learning, stable training of artificial neural networks with millions or billions of parameters is achieved using automatic differentiation frameworks such as JAX and Pytorch. Because these frameworks provide accelerated numerical linear algebra with automatic gradient tracking, they also enable differentiable implementations of numerical methods to be built. This facilitates faster gradient-based optimisation of geometry and materials, as well as solution of inverse problems. We demonstrate JAX-BEM, a differentiable Boundary Element Method (BEM) solver, showing that it matches the error of existing BEM codes for a benchmark problem and enables gradient-based geometry optimisation. Although the demonstrated examples are for acoustic simulations, the concept could be readily extended to electromagnetic waves.
title JAX-BEM: Gradient-Based Acoustic Shape Optimisation via a Differentiable Boundary Element Method
topic Computational Engineering, Finance, and Science
Computational Physics
url https://arxiv.org/abs/2604.21431