BEAMZ is an electromagnetic simulation package using the FDTD method. It features a high-level API for fast prototyping with just a few lines of code as well as an inverse design module for topology optimization using the adjoint method with Jax-based autodiff. Made for (but not limited to) photonic integrated circuits.

pip install beamz