chore: remove duplicate autograd modules

tidy3d/web/api/autograd/autograd.py

@@ -5,7 +5,6 @@
 from os.path import dirname
 from pathlib import Path
 
-import numpy as np
 from autograd.builtins import dict as dict_ag
 from autograd.extend import defvjp, primitive
 
@@ -47,16 +46,6 @@
 )
 from .forward import postprocess_fwd as _postprocess_fwd_impl
 from .forward import setup_fwd as _setup_fwd_impl
-from .io_utils import (
-    get_vjp_traced_fields as _get_vjp_traced_fields_impl,
-)
-from .io_utils import (
-    upload_sim_fields_keys as _upload_sim_fields_keys_impl,
-)
-
-# if True, will plot the adjoint fields on the plane provided. used for debugging only
-_INSPECT_ADJOINT_FIELDS = False
-_INSPECT_ADJOINT_PLANE = td.Box(center=(0, 0, 0), size=(td.inf, td.inf, 0))
 
 
 def is_valid_for_autograd(simulation: td.Simulation) -> bool:
@@ -623,21 +612,6 @@ def postprocess_fwd(
     )
 
 
-def upload_sim_fields_keys(sim_fields_keys: list[tuple], task_id: str, verbose: bool = False):
-    """Upload traced simulation field keys for adjoint runs (delegated)."""
-    return _upload_sim_fields_keys_impl(
-        sim_fields_keys=sim_fields_keys, task_id=task_id, verbose=verbose
-    )
-
-
-""" VJP maker for ADJ pass."""
-
-
-def get_vjp_traced_fields(task_id_adj: str, verbose: bool) -> AutogradFieldMap:
-    """Fetch VJP traced fields for a completed adjoint job (delegated)."""
-    return _get_vjp_traced_fields_impl(task_id_adj=task_id_adj, verbose=verbose)
-
-
 def _run_bwd(
     data_fields_original: AutogradFieldMap,
     sim_fields_original: AutogradFieldMap,
@@ -919,22 +893,6 @@ def setup_adj(
     )
 
 
-def _compute_eps_array(medium, frequencies):
-    """Deprecated shim, kept for backward compatibility; use ops_backward._compute_eps_array."""
-    from .backward import _compute_eps_array as __impl
-
-    return __impl(medium, frequencies)
-
-
-def _slice_field_data(
-    field_data: dict, freqs: np.ndarray, component_indicator: typing.Optional[str] = None
-) -> dict:
-    """Deprecated shim, kept for backward compatibility; use ops_backward._slice_field_data."""
-    from .backward import _slice_field_data as __impl
-
-    return __impl(field_data, freqs, component_indicator)
-
-
 def postprocess_adj(
     sim_data_adj: td.SimulationData,
     sim_data_orig: td.SimulationData,

tidy3d/web/api/autograd/backward.py

@@ -10,6 +10,7 @@
 from tidy3d.components.autograd.constants import ADJOINT_FREQ_CHUNK_SIZE
 from tidy3d.components.autograd.derivative_utils import DerivativeInfo
 from tidy3d.exceptions import AdjointError
+from tidy3d.packaging import disable_local_subpixel
 
 from .utils import E_to_D, get_derivative_maps
 
@@ -98,6 +99,7 @@ def _slice_field_data(
         return {k: v.sel(f=freqs) for k, v in field_data.items()}
 
 
+@disable_local_subpixel
 def postprocess_adj(
     sim_data_adj: td.SimulationData,
     sim_data_orig: td.SimulationData,