jax_sn.operators¶
This submodule defines the operators in the SN discretisation.
Each operator is defined by data (domain, quadrature, sweep ordering, fem matrices) and implementation strategies.
The data is bundled in `OperatorData`, split into differentiable and non-differentiable data.
Strategies are built either from combining operators or from a fused strategy.
The operators are all Lineax operators, which are built from a combination of data and strategy.
Subpackages¶
- jax_sn.operators.mass_operator
- jax_sn.operators.moment_operator
- jax_sn.operators.multi_group_operator
- jax_sn.operators.scatter_operator
- jax_sn.operators.single_group_operator
SingleGroupDataSingleGroupStrategyBaseSingleGroupStrategySingleGroupChainStrategySingleGroupSourceStrategySingleGroupOperatorSingleGroupChainOperatorSingleGroupSourceOperatorComposedSingleGroupStrategyComposedSingleGroupChainStrategyComposedSingleGroupSourceStrategysingle_group_moment_structure()single_group_angle_structure()- Submodules
- jax_sn.operators.transport_operator
SweepDataBasicSweepDataLeveledSweepDataLevelBasedParallelismProvidesLevelsTransportDataAbstractTransportOperatorTransportOperatorSingleAngleTransportOperatorMultiAngleSingleAngleForwardStrategyMultiAngleForwardStrategyComposedMultiAngleForwardStrategyAbstractInversionStrategySingleAngleInversionStrategyMultiAngleInversionStrategyComposedMultiAngleInversionStrategyVmapForwardBatchedForwardVmapSweepBatchedSweepVmapAnglesInversionScanAnglesInversionVmapAnglesForwardScanAnglesForwardVmapPallasSweepBatchedPallasSweepWarpSequentialSweepWarpLevelSweepWarpSequentialAnglesWarpLevelAngles- Subpackages
- Submodules
Submodules¶
- jax_sn.operators.base_linear_operator module
- Why not inheriting and implementing a forward?
AbstractLinearOperator_snDirectlyInvertibleOperatorDirectInverseSolverBaseOperatorDataTransposableOperatorDataStrategyTransposableStrategySymmetricStrategyDataStrategyLinearOperatorForwardDataStrategyLinearOperatorSymmetricDataStrategyLinearOperatorDataTransposeDataStrategyLinearOperatorTransposeBoolDataStrategyLinearOperator
- jax_sn.operators.operator_data module
- jax_sn.operators.sweep_data module