jax_sn¶ Subpackages¶ jax_sn.analytical_solutions psi_isotropic_box() Submodules jax_sn.domain CrossSectionData CrossSectionDataSingleGroup CrossSectionsIndexed GeometricDomain Domain Submodules jax_sn.energy_set compute_overlap_matrix() Submodules jax_sn.operator_creation SpatialBuilder DiscretisationBuilder SweptBuilder MultiGroupBuilder SingleGroupBuilder SingleGroupChainBuilder SingleGroupSourceBuilder MassBuilder MomentBuilder DiscreteBuilder ScatterBuilder SingleAngleTransportBuilder MultiAngleTransportBuilder DefaultMultiGroupStrategy Submodules jax_sn.operators Subpackages Submodules jax_sn.quadrature_set QuadratureSetBase QuadratureSet QuadratureSetReduced create_tn_quadrature_set() compute_3d_yklm_real_from_quadrature_set() compute_3d_wk_yklm_real_from_quadrature_set() Submodules jax_sn.solution_domain Element PrecomputedGeometricData FEMMatrices SolutionDomain SpatialData compute_setup_for_solution_domain() BasixLagrangianSimplex Submodules jax_sn.solvers MultiGroupLowerTriangularBlockSolver SolverDiagnostics RightHandSide IsotropicSourceSingleGroup TotalSource solve_multigroup_operator_rhs() Submodules jax_sn.tools assert_pytree_allclose() ParametrisedModule Submodules jax_sn.variance_reduction fw_cadis_scalar_flux() solve_continuous_adjoint() Submodules