CuWave

CuWave is a single-GPU, differentiable finite difference wave propagation code. Possible applications include

nondestructive testing via full waveform inversion
transient acoustic topology optimization
analog neural networks
transient photonic topology optimization

Documentation

• see the documentation for how the code works (AI-assisted; verify with sources for critical details)
• see examples for how to apply the code

Development status

Scalar & acoustic wave equations have been developed over the last 2 years and are thoroughly validated.

Important

Elastic & electromagnetic wave equations were developed with AI assistance (Claude) and have undergone less validation. The elastic wave equation is currently being validated against experimental results.

Performance

CuWave's runtime for identical discretizations is comparable to that of other established wave propagation finite difference codes. Speedups in 2D with reflecting boundaries (a ratio above 1 means CuWave is that many times faster) compared to the following frameworks:

• scalar wave equation
  • Deepwave (forward: ~1.1x, sensitivity: ~1x)
  • NVIDIA Warp (forward: ~2.1x, sensitivity: ~1.8x)
  • SeismicWaves.jl (forward: ~4.4x, sensitivity: ~4x)
  • Devito on CPU (forward: ~3.6x, sensitivity: ~3.8x)
• elastic wave equation
  • Deepwave (forward: ~1.1x, sensitivity: ~1.3x)
  • SeismicWaves.jl (forward: ~4x, sensitivity: ~5x) Tested on one NVIDIA RTX PRO 500 Blackwell laptop GPU (6 GB) on the largest possible grids with CuWave's superposition_sensitivity as reference for the sensitivities. The specific numbers need to be taken with a grain of salt, as they are subject to specific hardware and simulation setup. All implementations operate on the same order of magnitude.

Additional benefits of CuWave are

• the built-in higher order finite difference schemes, allowing for fewer grid points
• a sensitivity analysis whose memory is independent of the number of timesteps, allowing for orders of magnitude larger grids

Install

pip install cuwave

Requires an NVIDIA GPU and CuPy matching your CUDA toolkit (e.g. pip install cupy-cuda12x), which is not pulled in automatically. For the full installation, including the optional PyTorch and running the tests, see docs/install.md.

References

If you use our code for your scientific research, please acknowledge this by referring to the following publication:

Herrmann, L., Bürchner, T., Kudela, L., Kollmannsberger, S., 2026, A memory-efficient adjoint method to enable billion parameter optimization on a single GPU in dynamic problems, Structural and Multidisciplinary Optimization, Volume 69, 52 (2026), DOI: 10.1007/s00158-025-04237-y

Contact

For questions, bug reports, or collaboration inquiries, please don't hesitate to contact Leon Herrmann at herrmann.leon@pm.me.

License

MIT; see LICENSE.