jaxmodel.blocks

jaxmodel.blocks.affine_eq_from_parts(var_spec: VariableSpec, y_blocks: Sequence[Tuple[Array, str]], rhs_const: Array | None = None, x_blocks: Sequence[Tuple[Array, str]] | None = None, name: str = 'affine_eq')

jaxmodel.blocks.affine_ineq_from_parts(var_spec: VariableSpec, y_blocks: Sequence[Tuple[Array, str]], rhs_const: Array | None = None, x_blocks: Sequence[Tuple[Array, str]] | None = None, name: str = 'affine_ineq')

jaxmodel.blocks.nonlinear_eq_from_parts(var_spec: VariableSpec, linear_y_terms: Sequence[Tuple[Array, str]] | None = None, nonlinear_y_terms: Sequence[Tuple[Array, Callable, str]] | None = None, rhs_const: Array | None = None, x_blocks: Sequence[Tuple[Array, str]] | None = None, x_direct_name: str | None = None, name: str = 'nonlinear_eq')

jaxmodel.blocks.nonlinear_ineq_from_parts(var_spec: VariableSpec, linear_y_terms: Sequence[Tuple[Array, str]] | None = None, nonlinear_y_terms: Sequence[Tuple[Array, Callable, str]] | None = None, rhs_const: Array | None = None, x_blocks: Sequence[Tuple[Array, str]] | None = None, name: str = 'nonlinear_ineq')