Skip to content

Delaunay JAX

Choose a tag to compare

@Jammy2211 Jammy2211 released this 15 Dec 10:10
· 264 commits to main since this release

The adaptive Delaunay mesh using a Hilbert image-mesh now supports fully JAX'd likelihood functions running on GPU, which was disabled in previous releases.

The Delaunay mesh itself is not computed on GPU, but CPU, via a JAX pure_callback. Full details are provided below, but this does not impact significantly on performance:

PyAutoLabs/PyAutoArray#199