tidy3d.plugins.adjoint.JaxMedium#

class JaxMedium[source]#

Bases: Medium, AbstractJaxMedium

A Medium registered with jax.

Parameters:
  • name (Optional[str] = None) – Optional unique name for medium.

  • frequency_range (Optional[Tuple[float, float]] = None) – [units = (Hz, Hz)]. Optional range of validity for the medium.

  • allow_gain (bool = False) – Allow the medium to be active. Caution: simulations with a gain medium are unstable, and are likely to diverge.Simulations where ‘allow_gain’ is set to ‘True’ will still be charged even if diverged. Monitor data up to the divergence point will still be returned and can be useful in some cases.

  • nonlinear_spec (Union[NonlinearSpec, NonlinearSusceptibility] = None) – Nonlinear spec applied on top of the base medium properties.

  • modulation_spec (Optional[ModulationSpec] = None) – Modulation spec applied on top of the base medium properties.

  • heat_spec (Union[FluidSpec, SolidSpec, NoneType] = None) – Specification of the medium heat properties. They are used for solving the heat equation via the HeatSimulation interface. Such simulations can be used for investigating the influence of heat propagation on the properties of optical systems. Once the temperature distribution in the system is found using HeatSimulation object, Simulation.perturbed_mediums_copy() can be used to convert mediums with perturbation models defined into spatially dependent custom mediums. Otherwise, the heat_spec does not directly affect the running of an optical Simulation.

  • permittivity (ConstrainedFloatValue = 1.0) – [units = None (relative permittivity)]. Relative permittivity.

  • conductivity (float = 0.0) – [units = S/um]. Electric conductivity. Defined such that the imaginary part of the complex permittivity at angular frequency omega is given by conductivity/omega.

  • permittivity_jax (Union[float, NumpyArrayType, Array, JVPTracer, object] = 1.0) – Relative permittivity of the medium. May be a jax Array.

  • conductivity_jax (Union[float, NumpyArrayType, Array, JVPTracer, object] = 0.0) – [units = S/um]. Electric conductivity. Defined such that the imaginary part of the complex permittivity at angular frequency omega is given by conductivity/omega.

Attributes

Methods

store_vjp(grad_data_fwd, grad_data_adj, ...)

Returns the gradient of the medium parameters given forward and adjoint field data.

permittivity_jax#
conductivity_jax#
store_vjp(grad_data_fwd, grad_data_adj, sim_bounds, wvl_mat, inside_fn)[source]#

Returns the gradient of the medium parameters given forward and adjoint field data.

__hash__()#

Hash method.