tidy3d.plugins.adjoint.JaxSimulationData#

class JaxSimulationData[source]#

Bases: SimulationData, JaxObject

A SimulationData registered with jax.

Parameters:

Attributes

grad_data_symmetry

self.grad_data but with symmetry_expanded_copy applied.

grad_eps_data_symmetry

self.grad_eps_data but with symmetry_expanded_copy applied.

monitor_data

Dictionary of .output_data monitor .name to the corresponding data.

output_monitor_data

Dictionary of .output_data monitor .name to the corresponding data.

data

log

Logging for Tidy3d.

Methods

from_sim_data(sim_data, jax_info[, task_id])

Construct a JaxSimulationData instance from a SimulationData.

make_adjoint_simulation(fwidth, run_time)

Make an adjoint simulation out of the data provided (generally, the vjp sim data).

normalize_adjoint_fields()

Make copy of jax_sim_data with grad_data (fields) normalized by adjoint sources.

split_data(mnt_data, jax_info)

Split list of monitor data into data, output_data, grad_data, and grad_eps_data.

split_fwd_sim_data(sim_data, jax_info)

Split a SimulationData into two parts, containing user and gradient data.

output_data#
grad_data#
grad_eps_data#
simulation#
task_id#
property grad_data_symmetry#

self.grad_data but with symmetry_expanded_copy applied.

property grad_eps_data_symmetry#

self.grad_eps_data but with symmetry_expanded_copy applied.

property output_monitor_data#

Dictionary of .output_data monitor .name to the corresponding data.

property monitor_data#

Dictionary of .output_data monitor .name to the corresponding data.

static split_data(mnt_data, jax_info)[source]#

Split list of monitor data into data, output_data, grad_data, and grad_eps_data.

classmethod from_sim_data(sim_data, jax_info, task_id=None)[source]#

Construct a JaxSimulationData instance from a SimulationData.

classmethod split_fwd_sim_data(sim_data, jax_info)[source]#

Split a SimulationData into two parts, containing user and gradient data.

make_adjoint_simulation(fwidth, run_time)[source]#

Make an adjoint simulation out of the data provided (generally, the vjp sim data).

normalize_adjoint_fields()[source]#

Make copy of jax_sim_data with grad_data (fields) normalized by adjoint sources.

__hash__()#

Hash method.