Adjoint

• Adjoint Plugin Developer Guide
  • Context
    • The Need for Simulation Differentiation
    • Numerical Derivatives
    • Automatic Differentiation
    • Adjoint Method
    • Adjoint Plugin
    • AD using Jax
    • Outline of Guide
  • Automatic Differentiation w/ Jax
    • Vector Jacobian Products (VJPs)
    • Custom VJPs
    • The VJP of web.run()
    • Where do derivatives come from?
  • Jax Nuts and Bolts
    • Registering Tidy3D components with jax
    • Jax Subclasses of Tidy3D objects
    • Conversion Between JaxTidy3D and Tidy3D Components
    • The One Exception: DataArrays
  • Adjoint Method
    • Gradient Monitors
    • Adjoint Source Creation
    • Adjoint Simulation Creation
    • PostProcessing Forward and Adjoint Fields for VJP Creation
  • Using the Plugin
    • Defining Jax Simulation
      • Input Structures
      • Output Monitors
      • Gradient Monitors
      • Other Differences
    • Adjoint web.run function
    • Working with JaxSimulationData
    • Putting it all together
  • Differentiating using jax
    • Derivative Functions
      • grad
      • value_and_grad
      • jax.numpy
  • Conclusion

tidy3d.plugins.adjoint.web.run	Submits a JaxSimulation to server, starts running, monitors progress, downloads, and loads results as a JaxSimulationData object.
tidy3d.plugins.adjoint.web.run_async	Submits a set of JaxSimulation objects to server, starts running, monitors progress, downloads, and loads results as a tuple of JaxSimulationData objects.
tidy3d.plugins.adjoint.JaxBox	A Box registered with jax.
tidy3d.plugins.adjoint.JaxPolySlab	A PolySlab registered with jax.
tidy3d.plugins.adjoint.JaxMedium	A Medium registered with jax.
tidy3d.plugins.adjoint.JaxAnisotropicMedium	A Medium registered with jax.
tidy3d.plugins.adjoint.JaxCustomMedium	A CustomMedium registered with jax.
tidy3d.plugins.adjoint.JaxStructure	A Structure registered with jax.
tidy3d.plugins.adjoint.JaxSimulation	A Simulation registered with jax.
tidy3d.plugins.adjoint.JaxSimulationData	A SimulationData registered with jax.
tidy3d.plugins.adjoint.JaxModeData	A ModeData registered with jax.
tidy3d.plugins.adjoint.JaxPermittivityDataset	A PermittivityDataset registered with jax.
tidy3d.plugins.adjoint.JaxDataArray	A DataArray-like class that only wraps xarray for jax compatibility.
tidy3d.plugins.adjoint.utils.filter.ConicFilter	Filter that convolves an image with a conical mask, used for larger feature sizes.
tidy3d.plugins.adjoint.utils.filter.BinaryProjector	Projects a grayscale image towards min and max values using a smooth tanh function.
tidy3d.plugins.adjoint.utils.penalty.RadiusPenalty	Generates a penalty for radius of curvature of set of points.