tidy3d.plugins.adjoint.web.run

class run

Bases:

Submits a JaxSimulation to server, starts running, monitors progress, downloads, and loads results as a JaxSimulationData object. Can be included within a function that will have jax.grad applied.

Parameters:

• simulation (JaxSimulation) – Simulation to upload to server.

• task_name (str) – Name of task.

• path (str = "simulation_data.hdf5") – Path to download results file (.hdf5), including filename.

• folder_name (str = "default") – Name of folder to store task on web UI.

• callback_url (str = None) – Http PUT url to receive simulation finish event. The body content is a json file with fields {'id', 'status', 'name', 'workUnit', 'solverVersion'}.

• verbose (bool = True) – If True, will print progressbars and status, otherwise, will run silently.

Returns:

Object containing solver results for the supplied JaxSimulation.

Return type:

JaxSimulationData

bwd(folder_name, path, callback_url, verbose, res, sim_data_vjp)

Run backward pass and return simulation storing vjp of the objective w.r.t. the sim.

fun(task_name, folder_name='default', path='simulation_data.hdf5', callback_url=None, verbose=True)

Submits a JaxSimulation to server, starts running, monitors progress, downloads, and loads results as a JaxSimulationData object. Can be included within a function that will have jax.grad applied.

Parameters:

• simulation (JaxSimulation) – Simulation to upload to server.

• task_name (str) – Name of task.

• path (str = "simulation_data.hdf5") – Path to download results file (.hdf5), including filename.

• folder_name (str = "default") – Name of folder to store task on web UI.

• callback_url (str = None) – Http PUT url to receive simulation finish event. The body content is a json file with fields {'id', 'status', 'name', 'workUnit', 'solverVersion'}.

• verbose (bool = True) – If True, will print progressbars and status, otherwise, will run silently.

Returns:

Object containing solver results for the supplied JaxSimulation.

Return type:

JaxSimulationData

fwd(task_name, folder_name, path, callback_url, verbose)

Run forward pass and stash extra objects for the backwards pass.

nondiff_argnums = (1, 2, 3, 4, 5)

symbolic_zeros = False