virtual_casing_jaxΒΆ
JAX implementation of the virtual casing principle with high-order singular quadrature.
Contents
- Overview
- Equations
- Numerics
- Functional API
- ComputeGradB and Autodiff
- Off-Surface Evaluation
- VMEC Extender
- Exterior Field API
- Differentiability
- Test Data
- Implementation
- Performance
- Benchmarks
- Using in SIMSOPT
- Validation
- Parity Datasets
- API
complete_vec_field()cross_prod()dot_prod()fft_c2r()fft_r2c()grad2d()normal_orientation()resample()rotate_toroidal()surf_normal_area_elem()upsample()biotsavart_fx_u()biotsavart_fxd_u()laplace_dx_u()laplace_fx_u()laplace_fxd2_u()laplace_fxd_u()OffsurfaceConvergenceErrorbiotsavart_fx_u_eval()biotsavart_fxd_u_eval()computeB_offsurface_adaptive()computeB_offsurface_adaptive_schedule()computeB_offsurface_baseline()computeGradB_offsurface_adaptive_schedule()curl_single_layer_gradient()field_period_target_coords()laplace_dx_u_eval()laplace_dx_u_eval_singular()laplace_fxd2_u_eval()laplace_fxd2_u_eval_singular()laplace_fxd2_u_eval_vec()laplace_fxd2_u_eval_vec_singular()laplace_fxd_u_eval()laplace_fxd_u_eval_singular()laplace_fxd_u_eval_vec()laplace_fxd_u_eval_vec_singular()FunctionalSetupbuild_patch_idx()build_quad_setup()build_surface_coord()compute_external_B_functional()compute_external_B_jvp_columns_functional()compute_external_B_normal_functional()compute_external_B_normal_jvp_columns_functional()compute_external_B_offsurf_functional()compute_external_gradB_functional()compute_external_gradB_offsurf_functional()compute_internal_B_functional()compute_internal_gradB_functional()prepare_functional_setup()select_patch_dim_from_geom()target_surface_normal()PrecisionPlanQuadSetupVirtualCasingJAXVirtualCasingB_cyl_from_B_xyz()ExteriorFieldConfigNearSurfaceTaylorPlanVirtualCasingExteriorFieldVmecSurfaceFieldDatabuild_near_surface_taylor_plan()cyl_to_xyz()xyz_vec_to_cyl_vec()surface_field_from_vmec_jax()evaluate_on_rphiz_grid()write_extended_field_netcdf()write_mgrid_like()boundary_normal_residual()exterior_B_magnitude_squared()exterior_Bn_squared()grid_divergence_penalty()Drand48SurfTypeVirtualCasingTestDatamagnetic_field_data()magnetic_field_data_offsurf()magnetic_field_grad_data()magnetic_field_grad_data_offsurf()surface_coordinates()SingularPrecompprecompute_singular()select_patch_dim()
- References