Index

A

ang_dim (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute) area_vector (virtual_casing_jax.exterior_field.VmecSurfaceFieldData attribute) AxisymCircleNarrow (virtual_casing_jax.testdata.SurfType attribute)

AxisymCircleWide (virtual_casing_jax.testdata.SurfType attribute) AxisymNarrow (virtual_casing_jax.testdata.SurfType attribute) AxisymWide (virtual_casing_jax.testdata.SurfType attribute)

B

B_and_derivatives_xyz() (virtual_casing_jax.exterior_field.VirtualCasingExteriorField method) B_cyl() (virtual_casing_jax.exterior_field.VirtualCasingExteriorField method) B_cyl_from_B_xyz() (in module virtual_casing_jax.exterior_field) B_external_xyz() (virtual_casing_jax.exterior_field.VirtualCasingExteriorField method) B_plasma_error_estimate() (virtual_casing_jax.exterior_field.VirtualCasingExteriorField method) B_plasma_near_surface_xyz() (virtual_casing_jax.exterior_field.VirtualCasingExteriorField method) B_plasma_on_surface() (virtual_casing_jax.exterior_field.VirtualCasingExteriorField method) B_plasma_xyz() (virtual_casing_jax.exterior_field.VirtualCasingExteriorField method) B_surface (virtual_casing_jax.exterior_field.NearSurfaceTaylorPlan attribute) B_total (virtual_casing_jax.exterior_field.VmecSurfaceFieldData attribute) B_xyz() (virtual_casing_jax.exterior_field.VirtualCasingExteriorField method)

B_xyz_sharded() (virtual_casing_jax.exterior_field.VirtualCasingExteriorField method) biotsavart_fx_u() (in module virtual_casing_jax.kernels) biotsavart_fx_u_eval() (in module virtual_casing_jax.integrals) biotsavart_fxd_u() (in module virtual_casing_jax.kernels) biotsavart_fxd_u_eval() (in module virtual_casing_jax.integrals) boundary_normal_residual() (in module virtual_casing_jax.objectives) branch (virtual_casing_jax.exterior_field.ExteriorFieldConfig attribute) build_near_surface_taylor_plan() (in module virtual_casing_jax.exterior_field) build_patch_idx() (in module virtual_casing_jax.functional) build_quad_setup() (in module virtual_casing_jax.functional) build_surface_coord() (in module virtual_casing_jax.functional)

C

center_rz (virtual_casing_jax.exterior_field.NearSurfaceTaylorPlan attribute) chunk_size (virtual_casing_jax.exterior_field.ExteriorFieldConfig attribute) complete_vec_field() (in module virtual_casing_jax.surface_ops) compute_external_B() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_B_autodiff() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_B_batch() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_B_functional() (in module virtual_casing_jax.functional) compute_external_B_jit() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_B_jvp_columns_functional() (in module virtual_casing_jax.functional) compute_external_B_normal_functional() (in module virtual_casing_jax.functional) compute_external_B_normal_jvp_columns_functional() (in module virtual_casing_jax.functional) compute_external_B_offsurf() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_B_offsurf_functional() (in module virtual_casing_jax.functional) compute_external_B_offsurf_schedule() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_B_offsurf_schedule_jit() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_gradB() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_gradB_batch() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_gradB_functional() (in module virtual_casing_jax.functional) compute_external_gradB_jit() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_gradB_offsurf() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_gradB_offsurf_functional() (in module virtual_casing_jax.functional) compute_external_gradB_offsurf_schedule() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method)

compute_external_gradB_offsurf_schedule_jit() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_B() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_B_batch() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_B_functional() (in module virtual_casing_jax.functional) compute_internal_B_jit() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_B_offsurf() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_B_offsurf_schedule() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_B_offsurf_schedule_jit() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_gradB() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_gradB_batch() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_gradB_functional() (in module virtual_casing_jax.functional) compute_internal_gradB_jit() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_gradB_offsurf() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_gradB_offsurf_schedule() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_gradB_offsurf_schedule_jit() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) computeB_offsurface_adaptive() (in module virtual_casing_jax.integrals) computeB_offsurface_adaptive_schedule() (in module virtual_casing_jax.integrals) computeB_offsurface_baseline() (in module virtual_casing_jax.integrals) computeGradB_offsurface_adaptive_schedule() (in module virtual_casing_jax.integrals) cross_prod() (in module virtual_casing_jax.surface_ops) curl_single_layer_gradient() (in module virtual_casing_jax.integrals) cyl_to_xyz() (in module virtual_casing_jax.exterior_field)

D

default_schedule_levels() (in module virtual_casing_jax.exterior_field) digits (virtual_casing_jax.exterior_field.ExteriorFieldConfig attribute) dot_prod() (in module virtual_casing_jax.surface_ops)

Drand48 (class in virtual_casing_jax.testdata) dtype (virtual_casing_jax.exterior_field.ExteriorFieldConfig attribute) dX (virtual_casing_jax.virtual_casing.QuadSetup attribute)

E

evaluate_on_rphiz_grid() (in module virtual_casing_jax.grid_export) export_rphiz_grid() (virtual_casing_jax.exterior_field.VirtualCasingExteriorField method)

exterior_B_magnitude_squared() (in module virtual_casing_jax.objectives) exterior_Bn_squared() (in module virtual_casing_jax.objectives) ExteriorFieldConfig (class in virtual_casing_jax.exterior_field)

F

fft_c2r() (in module virtual_casing_jax.surface_ops) fft_r2c() (in module virtual_casing_jax.surface_ops)

field_period_target_coords() (in module virtual_casing_jax.integrals) from_vmec() (virtual_casing_jax.simsopt_virtual_casing.VirtualCasing class method) FunctionalSetup (class in virtual_casing_jax.functional)

G

gamma (virtual_casing_jax.exterior_field.VmecSurfaceFieldData attribute) Gpou (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute) grad2d() (in module virtual_casing_jax.surface_ops)

gradB_plasma_xyz() (virtual_casing_jax.exterior_field.VirtualCasingExteriorField method) gradB_surface (virtual_casing_jax.exterior_field.NearSurfaceTaylorPlan attribute) gradB_xyz() (virtual_casing_jax.exterior_field.VirtualCasingExteriorField method) grid_divergence_penalty() (in module virtual_casing_jax.objectives)

H

half_period (virtual_casing_jax.functional.FunctionalSetup attribute)

hedgehog_order (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute) hedgehog_wts (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute)

I

I_G2P (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute)

interp_idx (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute)

L

laplace_dx_u() (in module virtual_casing_jax.kernels) laplace_dx_u_eval() (in module virtual_casing_jax.integrals) laplace_dx_u_eval_singular() (in module virtual_casing_jax.integrals) laplace_fx_u() (in module virtual_casing_jax.kernels) laplace_fxd2_u() (in module virtual_casing_jax.kernels) laplace_fxd2_u_eval() (in module virtual_casing_jax.integrals) laplace_fxd2_u_eval_singular() (in module virtual_casing_jax.integrals) laplace_fxd2_u_eval_vec() (in module virtual_casing_jax.integrals) laplace_fxd2_u_eval_vec_singular() (in module virtual_casing_jax.integrals)

laplace_fxd_u() (in module virtual_casing_jax.kernels) laplace_fxd_u_eval() (in module virtual_casing_jax.integrals) laplace_fxd_u_eval_singular() (in module virtual_casing_jax.integrals) laplace_fxd_u_eval_vec() (in module virtual_casing_jax.integrals) laplace_fxd_u_eval_vec_singular() (in module virtual_casing_jax.integrals) level_sources() (virtual_casing_jax.exterior_field.VirtualCasingExteriorField method) levels (virtual_casing_jax.exterior_field.ExteriorFieldConfig attribute) LHD (virtual_casing_jax.testdata.SurfType attribute) load() (virtual_casing_jax.simsopt_virtual_casing.VirtualCasing class method)

M

M_G2P (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute) magnetic_field_data() (in module virtual_casing_jax.testdata) (virtual_casing_jax.testdata.VirtualCasingTestData static method) magnetic_field_data_offsurf() (in module virtual_casing_jax.testdata) (virtual_casing_jax.testdata.VirtualCasingTestData static method) magnetic_field_grad_data() (in module virtual_casing_jax.testdata) (virtual_casing_jax.testdata.VirtualCasingTestData static method) magnetic_field_grad_data_offsurf() (in module virtual_casing_jax.testdata) (virtual_casing_jax.testdata.VirtualCasingTestData static method) module virtual_casing_jax.exterior_field virtual_casing_jax.functional virtual_casing_jax.grid_export virtual_casing_jax.integrals virtual_casing_jax.kernels virtual_casing_jax.objectives virtual_casing_jax.simsopt_virtual_casing virtual_casing_jax.singular_quadrature virtual_casing_jax.surface_ops virtual_casing_jax.testdata virtual_casing_jax.virtual_casing

N

NearSurfaceTaylorPlan (class in virtual_casing_jax.exterior_field) nfp (virtual_casing_jax.exterior_field.NearSurfaceTaylorPlan attribute) (virtual_casing_jax.exterior_field.VmecSurfaceFieldData attribute) (virtual_casing_jax.functional.FunctionalSetup attribute) nfp_eff (virtual_casing_jax.functional.FunctionalSetup attribute)

ngrid (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute) NoneType (virtual_casing_jax.testdata.SurfType attribute) normal (virtual_casing_jax.exterior_field.VmecSurfaceFieldData attribute) (virtual_casing_jax.virtual_casing.QuadSetup attribute) normal_orientation() (in module virtual_casing_jax.surface_ops) npolar (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute)

O

OffsurfaceConvergenceError

orient (virtual_casing_jax.functional.FunctionalSetup attribute) (virtual_casing_jax.virtual_casing.QuadSetup attribute)

P

patch_dim (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute) patch_dim0 (virtual_casing_jax.functional.FunctionalSetup attribute) (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute) (virtual_casing_jax.virtual_casing.PrecisionPlan attribute) patch_idx (virtual_casing_jax.functional.FunctionalSetup attribute) (virtual_casing_jax.virtual_casing.PrecisionPlan attribute) patch_idx_cache (virtual_casing_jax.virtual_casing.QuadSetup attribute) phi (virtual_casing_jax.exterior_field.VmecSurfaceFieldData attribute)

plan_near_surface() (virtual_casing_jax.exterior_field.VirtualCasingExteriorField method) plan_precision() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) plan_surface_precision() (virtual_casing_jax.exterior_field.VirtualCasingExteriorField method) plot() (virtual_casing_jax.simsopt_virtual_casing.VirtualCasing method) Ppou (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute) PrecisionPlan (class in virtual_casing_jax.virtual_casing) precompute_singular() (in module virtual_casing_jax.singular_quadrature) prepare_functional_setup() (in module virtual_casing_jax.functional)

Q

quad_coord (virtual_casing_jax.virtual_casing.QuadSetup attribute) quad_np (virtual_casing_jax.functional.FunctionalSetup attribute) (virtual_casing_jax.virtual_casing.PrecisionPlan attribute) (virtual_casing_jax.virtual_casing.QuadSetup attribute) quad_nt (virtual_casing_jax.functional.FunctionalSetup attribute) (virtual_casing_jax.virtual_casing.PrecisionPlan attribute) (virtual_casing_jax.virtual_casing.QuadSetup attribute)

QuadSetup (class in virtual_casing_jax.virtual_casing) Quas3 (virtual_casing_jax.testdata.SurfType attribute) qw (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute) qx (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute)

R

rad_dim (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute) rad_dim_base (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute) rand() (virtual_casing_jax.testdata.Drand48 method)

resample() (in module virtual_casing_jax.surface_ops) rotate_toroidal() (in module virtual_casing_jax.surface_ops) RotatingEllipseNarrow (virtual_casing_jax.testdata.SurfType attribute) RotatingEllipseWide (virtual_casing_jax.testdata.SurfType attribute)

S

save() (virtual_casing_jax.simsopt_virtual_casing.VirtualCasing method) select_patch_dim() (in module virtual_casing_jax.singular_quadrature) select_patch_dim_from_geom() (in module virtual_casing_jax.functional) setup() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) signgs (virtual_casing_jax.exterior_field.VmecSurfaceFieldData attribute) SingularPrecomp (class in virtual_casing_jax.singular_quadrature) source_convention (virtual_casing_jax.exterior_field.VmecSurfaceFieldData attribute) source_field_scale() (in module virtual_casing_jax.integrals) src_np (virtual_casing_jax.functional.FunctionalSetup attribute) src_nphi (virtual_casing_jax.exterior_field.ExteriorFieldConfig attribute) src_nt (virtual_casing_jax.functional.FunctionalSetup attribute)

src_ntheta (virtual_casing_jax.exterior_field.ExteriorFieldConfig attribute) state (virtual_casing_jax.testdata.Drand48 attribute) Stell (virtual_casing_jax.testdata.SurfType attribute) stellsym (virtual_casing_jax.exterior_field.VmecSurfaceFieldData attribute) surf_normal_area_elem() (in module virtual_casing_jax.surface_ops) surf_np (virtual_casing_jax.functional.FunctionalSetup attribute) surf_nt (virtual_casing_jax.functional.FunctionalSetup attribute) surface_coordinates() (in module virtual_casing_jax.testdata) (virtual_casing_jax.testdata.VirtualCasingTestData static method) surface_rz (virtual_casing_jax.exterior_field.NearSurfaceTaylorPlan attribute) SurfType (class in virtual_casing_jax.testdata)

T

target_chunk_size (virtual_casing_jax.exterior_field.ExteriorFieldConfig attribute) target_surface_normal() (in module virtual_casing_jax.functional)

theta (virtual_casing_jax.exterior_field.VmecSurfaceFieldData attribute) trg_np (virtual_casing_jax.functional.FunctionalSetup attribute) trg_nt (virtual_casing_jax.functional.FunctionalSetup attribute)

U

upsample() (in module virtual_casing_jax.surface_ops)

use_jit_schedule (virtual_casing_jax.exterior_field.ExteriorFieldConfig attribute)

V

validate_orientation (virtual_casing_jax.exterior_field.ExteriorFieldConfig attribute) virtual_casing_jax.exterior_field module virtual_casing_jax.functional module virtual_casing_jax.grid_export module virtual_casing_jax.integrals module virtual_casing_jax.kernels module virtual_casing_jax.objectives module virtual_casing_jax.simsopt_virtual_casing module

virtual_casing_jax.singular_quadrature module virtual_casing_jax.surface_ops module virtual_casing_jax.testdata module virtual_casing_jax.virtual_casing module VirtualCasing (class in virtual_casing_jax.simsopt_virtual_casing) VirtualCasingExteriorField (class in virtual_casing_jax.exterior_field) VirtualCasingJAX (class in virtual_casing_jax.virtual_casing) VirtualCasingTestData (class in virtual_casing_jax.testdata) VmecSurfaceFieldData (class in virtual_casing_jax.exterior_field)

W

W7X (virtual_casing_jax.testdata.SurfType attribute) W7X_ (virtual_casing_jax.testdata.SurfType attribute)

write_extended_field_netcdf() (in module virtual_casing_jax.grid_export) write_mgrid_like() (in module virtual_casing_jax.grid_export)

X

xyz_vec_to_cyl_vec() (in module virtual_casing_jax.exterior_field)