jax_sbgeom package
jax-sbgeom: A JAX-based package for geometric operations.
Subpackages
- jax_sbgeom.coils package
CoilFiniteSizeCoilCentroidFrameFrenetSerretFrameRadialVectorFrameRotationMinimizedFrameCoilSetFiniteSizeCoilSetFourierCoilconvert_to_fourier_coilset()convert_to_fourier_coil()convert_fourier_coil_to_equal_arclength()convert_fourier_coilset_to_equal_arclength()DiscreteCoilmesh_coil_surface()mesh_coilset_surface()mesh_coil_volumetric()create_coil_winding_surface()create_optimized_coil_winding_surface()calculate_normals_from_closest_point_on_mesh()- Submodules
- jax_sbgeom.flux_surfaces package
FluxSurfaceFluxSurfaceDataFluxSurfaceModesFluxSurfaceSettingsToroidalExtentFluxSurfaceBaseParametrisedSurfaceFluxSurfaceNormalExtendedFluxSurfaceNormalExtendedNoPhiFluxSurfaceNormalExtendedConstantPhiFluxSurfaceFourierExtendedFluxSurfaceExtendedDistanceMatrixmesh_surface()mesh_surfaces_closed()mesh_watertight_layers()mesh_tetrahedra()create_fourier_representation()convert_to_different_settings()convert_to_equal_arclength()create_fourier_representation_d_interp()create_fourier_representation_d_interp_equal_arclength()create_flux_surface_d_interp()create_flux_surface_d_interp_equal_arclength()create_extended_flux_surface_d_interp()create_extended_flux_surface_d_interp_equal_arclength()generate_thickness_matrix()- Submodules
- jax_sbgeom.interfaces package
- jax_sbgeom.jax_utils package
stack_jacfwd()interpolate_fractions()interpolate_array()interp1d_jax()interpolate_fractions_modulo()interpolate_array_modulo()interpolate_array_modulo_broadcasted()bilinear_interp()cumulative_trapezoid_uniform_periodic()pchip_interpolation()resample_uniform_periodic_linear()resample_uniform_periodic_pchip()reverse_except_begin()mesh_to_pyvista_mesh()surface_normals_from_mesh()vertices_to_pyvista_polyline()boundary_centroids_from_tetrahedron()boundary_normal_vectors_from_tetrahedron()- Submodules