sheap.Minimizer.LB_nonlinear module

make_build_linear_system_from_phys_profile(fused_profile)[source]

Same semantics; column builder is now vmapped over a static arange so JAX can compile it once and cache the result.

Parameters:: fused_profile (Callable)

build_varpro_loss_from_profile_and_params_obj(fused_profile, params_obj, nonlinear_raw_idx, linear_phys_idx=None, weighted=True, lambda_reg=0.0, reg_matrix=None)[source]

Parameters:

fused_profile (Callable)
nonlinear_raw_idx (jax.numpy.ndarray)
linear_phys_idx (jax.numpy.ndarray | None)
weighted (bool)
lambda_reg (float)
reg_matrix (jax.numpy.ndarray | None)