Index

A

A (jaxmodel.approximations.LinearizationData attribute) (jaxmodel.LinearizationData attribute) abs() (kkthn.builder.KKTHardNet method) add() (kkthn.builder.ConstraintList method) add_affine_equality() (jaxmodel.highlevel.HighLevelNLPBuilder method) (jaxmodel.HighLevelNLPBuilder method) add_affine_inequality() (jaxmodel.highlevel.HighLevelNLPBuilder method) (jaxmodel.HighLevelNLPBuilder method) add_block() (jaxmodel.VariableBuilder method) (jaxmodel.variables.VariableBuilder method) add_constraint() (jaxmodel.builder.NLPBuilder method) (jaxmodel.NLPBuilder method) add_constraints() (jaxmodel.builder.NLPBuilder method) (jaxmodel.NLPBuilder method) add_eq_block() (jaxmodel.builder.NLPBuilder method) (jaxmodel.NLPBuilder method) add_eq_scalar() (jaxmodel.builder.NLPBuilder method) (jaxmodel.NLPBuilder method) add_ineq_block() (jaxmodel.builder.NLPBuilder method) (jaxmodel.NLPBuilder method) add_ineq_scalar() (jaxmodel.builder.NLPBuilder method) (jaxmodel.NLPBuilder method) add_inverse_parameter() (kkthn.builder.KKTHardNet method) add_matrix() (jaxmodel.VariableBuilder method) (jaxmodel.variables.VariableBuilder method) add_nonlinear_equality() (jaxmodel.highlevel.HighLevelNLPBuilder method) (jaxmodel.HighLevelNLPBuilder method)

add_nonlinear_inequality() (jaxmodel.highlevel.HighLevelNLPBuilder method) (jaxmodel.HighLevelNLPBuilder method) add_parameter() (jaxmodel.highlevel.HighLevelNLPBuilder method) (jaxmodel.HighLevelNLPBuilder method) (kkthn.builder.KKTHardNet method) add_quadratic_equality() (jaxmodel.highlevel.HighLevelNLPBuilder method) (jaxmodel.HighLevelNLPBuilder method) add_quadratic_inequality() (jaxmodel.highlevel.HighLevelNLPBuilder method) (jaxmodel.HighLevelNLPBuilder method) add_scalar() (jaxmodel.VariableBuilder method) (jaxmodel.variables.VariableBuilder method) add_tensor() (jaxmodel.VariableBuilder method) (jaxmodel.variables.VariableBuilder method) add_variable() (jaxmodel.highlevel.HighLevelNLPBuilder method) (jaxmodel.HighLevelNLPBuilder method) (kkthn.builder.KKTHardNet method) add_vector() (jaxmodel.VariableBuilder method) (jaxmodel.variables.VariableBuilder method) affine_eq_from_parts() (in module jaxmodel) (in module jaxmodel.blocks) affine_ineq_from_parts() (in module jaxmodel) (in module jaxmodel.blocks) aggregate_constraint_jacobian() (in module jaxmodel.constraints) aggregate_constraints() (in module jaxmodel.constraints) armijo_alpha (kkthn.projection.ProjectionSettings attribute) armijo_beta (kkthn.projection.ProjectionSettings attribute) armijo_max_steps (kkthn.projection.ProjectionSettings attribute)

B

b (jaxmodel.approximations.LinearizationData attribute) (jaxmodel.LinearizationData attribute) backward_reg (kkthn.projection.ProjectionSettings attribute) batch_exp() (kkthn.builder.KKTHardNet method) batch_lin() (kkthn.builder.KKTHardNet method) batch_quad() (kkthn.builder.KKTHardNet method) batch_size (kkthn.training.KKTTrainConfig attribute) bound_residuals() (jaxmodel.JaxNLPModel method) (jaxmodel.model.JaxNLPModel method) BoundSpec (class in jaxmodel) (class in jaxmodel.bounds) build() (jaxmodel.builder.NLPBuilder method) (jaxmodel.highlevel.HighLevelNLPBuilder method) (jaxmodel.HighLevelNLPBuilder method) (jaxmodel.NLPBuilder method) (jaxmodel.VariableBuilder method) (jaxmodel.variables.VariableBuilder method)

build_diag_hess_y_objective() (in module jaxmodel.autodiff) build_grad_y_objective() (in module jaxmodel.autodiff) build_hess_y_objective() (in module jaxmodel.autodiff) build_jac_y_vector_fun() (in module jaxmodel.autodiff) build_model_from_string_problem() (in module kkthn.string_problem) build_sqp_data() (in module jaxmodel.approximations) build_sqp_subproblem_data() (in module jaxmodel.approximations)

C

C (jaxmodel.approximations.LinearizationData attribute) c (jaxmodel.approximations.QuadraticObjectiveData attribute) C (jaxmodel.LinearizationData attribute) c (jaxmodel.objective.QuadraticObjective attribute) (jaxmodel.QuadraticObjective attribute) (jaxmodel.QuadraticObjectiveData attribute) compile_native_projection() (in module kkthn.native_projection) cons_alpha (kkthn.training.KKTTrainConfig attribute) Constant (class in kkthn.builder)

constant (jaxmodel.objective.QuadraticObjective attribute) (jaxmodel.QuadraticObjective attribute) Constraint (class in kkthn.builder) ConstraintEntry (class in jaxmodel) (class in jaxmodel.constraints) ConstraintList (class in kkthn.builder) constraints (jaxmodel.approximations.SQPSubproblemData attribute) (jaxmodel.SQPSubproblemData attribute) cos() (kkthn.builder.KKTHardNet method)

D

d (jaxmodel.approximations.LinearizationData attribute) (jaxmodel.LinearizationData attribute) dataset() (kkthn.builder.KKTHardNet method) dataset_spec (kkthn.builder.KKTHardNet attribute) DatasetSpec (class in kkthn.builder) diag_hess_from_flat() (jaxmodel.objective.QuadraticObjective method) (jaxmodel.QuadraticObjective method)

diag_hess_y_objective() (jaxmodel.JaxNLPModel method) (jaxmodel.model.JaxNLPModel method) dims (kkthn.projection.ProjectionLayer attribute) drop_last (kkthn.training.KKTTrainConfig attribute) dtype (jaxmodel.builder.NLPBuilder attribute) (jaxmodel.NLPBuilder attribute) (kkthn.training.KKTTrainConfig attribute)

E

epoch_mlp (kkthn.training.KKTTrainConfig attribute) epochs (kkthn.training.KKTTrainConfig attribute) eq_block() (in module jaxmodel) (in module jaxmodel.constraints) eq_constraints (kkthn.projection.ProjectionLayer attribute) eq_residual() (jaxmodel.JaxNLPModel method) (jaxmodel.model.JaxNLPModel method) eq_scalar() (in module jaxmodel) (in module jaxmodel.constraints)

equality() (kkthn.builder.ConstraintList method) estimate() (kkthn.builder.KKTHardNet method) eta (kkthn.training.KKTTrainConfig attribute) eval() (kkthn.builder.Expression method) (kkthn.builder.VectorExpression method) exp() (kkthn.builder.KKTHardNet method) Expression (class in kkthn.builder) extract() (kkthn.builder.KKTHardNet method)

F

fb_eps (kkthn.projection.ProjectionSettings attribute) from_example() (jaxmodel.parameters.ParameterSpec static method) (jaxmodel.ParameterSpec static method)

fun (jaxmodel.ConstraintEntry attribute) (jaxmodel.constraints.ConstraintEntry attribute)

G

g_val (jaxmodel.approximations.LinearizationData attribute) (jaxmodel.LinearizationData attribute) glorot_init() (in module kkthn.backbone) gn_max_iters (kkthn.projection.ProjectionSettings attribute) gn_reg (kkthn.projection.ProjectionSettings attribute) gn_tol (kkthn.projection.ProjectionSettings attribute)

grad_f (jaxmodel.approximations.QuadraticObjectiveData attribute) (jaxmodel.QuadraticObjectiveData attribute) grad_from_flat() (jaxmodel.objective.QuadraticObjective method) (jaxmodel.QuadraticObjective method) grad_y_objective() (jaxmodel.JaxNLPModel method) (jaxmodel.model.JaxNLPModel method)

H

h_val (jaxmodel.approximations.LinearizationData attribute) (jaxmodel.LinearizationData attribute) hess_from_flat() (jaxmodel.objective.QuadraticObjective method) (jaxmodel.QuadraticObjective method) hess_y_objective() (jaxmodel.JaxNLPModel method) (jaxmodel.model.JaxNLPModel method)

hidden_layers (kkthn.training.KKTTrainConfig attribute) hidden_size (kkthn.training.KKTTrainConfig attribute) HighLevelNLPBuilder (class in jaxmodel) (class in jaxmodel.highlevel)

I

ineq_block() (in module jaxmodel) (in module jaxmodel.constraints) ineq_constraints (kkthn.projection.ProjectionLayer attribute) ineq_residual() (jaxmodel.JaxNLPModel method) (jaxmodel.model.JaxNLPModel method)

ineq_scalar() (in module jaxmodel) (in module jaxmodel.constraints) inequality() (kkthn.builder.ConstraintList method) init_mlp_params() (in module kkthn.backbone) inverse_parameter_names (kkthn.builder.KKTHardNet attribute) items (kkthn.builder.ConstraintList attribute)

J

jac_y_eq() (jaxmodel.JaxNLPModel method) (jaxmodel.model.JaxNLPModel method) jac_y_fun (jaxmodel.ConstraintEntry attribute) (jaxmodel.constraints.ConstraintEntry attribute) jac_y_ineq() (jaxmodel.JaxNLPModel method) (jaxmodel.model.JaxNLPModel method) jaxmodel module jaxmodel.approximations module jaxmodel.autodiff module jaxmodel.blocks module jaxmodel.bounds module jaxmodel.builder module

jaxmodel.constraints module jaxmodel.highlevel module jaxmodel.model module jaxmodel.objective module jaxmodel.parameters module jaxmodel.types module jaxmodel.variables module JaxNLPModel (class in jaxmodel) (class in jaxmodel.model)

K

kind (jaxmodel.ConstraintEntry attribute) (jaxmodel.constraints.ConstraintEntry attribute) (kkthn.builder.Constraint attribute) KKTHardNet (class in kkthn.builder) kkthn.backbone module kkthn.builder module

kkthn.native_projection module kkthn.projection module kkthn.string_problem module kkthn.training module KKTTrainConfig (class in kkthn.training)

L

l (jaxmodel.approximations.SQPSubproblemData attribute) (jaxmodel.SQPSubproblemData attribute) learning_rate (kkthn.training.KKTTrainConfig attribute) lin() (kkthn.builder.KKTHardNet method) linearization_data() (jaxmodel.JaxNLPModel method) (jaxmodel.model.JaxNLPModel method) LinearizationData (class in jaxmodel) (class in jaxmodel.approximations) linearize_constraints() (in module jaxmodel.approximations) (jaxmodel.JaxNLPModel method) (jaxmodel.model.JaxNLPModel method)

linearize_constraints_data() (in module jaxmodel.approximations) load() (kkthn.builder.KKTHardNet method) load_model_weights() (in module kkthn.training) load_native_projection() (in module kkthn.native_projection) load_or_compile_native_projection() (in module kkthn.native_projection) log() (kkthn.builder.KKTHardNet method) lower_bounds() (jaxmodel.JaxNLPModel method) (jaxmodel.model.JaxNLPModel method) lower_fun (jaxmodel.bounds.BoundSpec attribute) (jaxmodel.BoundSpec attribute)

M

make_batched_mlp_apply() (in module kkthn.backbone) make_projection_layer() (in module kkthn.projection) make_scalar_expr_fn() (in module kkthn.string_problem) matrix() (kkthn.builder.KKTHardNet method) max_backtrack_iter (kkthn.projection.ProjectionSettings attribute) metadata (jaxmodel.ConstraintEntry attribute) (jaxmodel.constraints.ConstraintEntry attribute) mlp_apply() (in module kkthn.backbone) model() (kkthn.builder.KKTHardNet method) module jaxmodel jaxmodel.approximations jaxmodel.autodiff jaxmodel.blocks jaxmodel.bounds jaxmodel.builder jaxmodel.constraints jaxmodel.highlevel jaxmodel.model jaxmodel.objective jaxmodel.parameters jaxmodel.types jaxmodel.variables kkthn.backbone kkthn.builder kkthn.native_projection kkthn.projection kkthn.string_problem kkthn.training

N

name (jaxmodel.ConstraintEntry attribute) (jaxmodel.constraints.ConstraintEntry attribute) names (jaxmodel.parameters.ParameterSpec attribute) (jaxmodel.ParameterSpec attribute) (jaxmodel.variables.VariableSpec attribute) (jaxmodel.VariableSpec attribute) NativeProjection (class in kkthn.native_projection)

newton_step_length (kkthn.projection.ProjectionSettings attribute) NLPBuilder (class in jaxmodel) (class in jaxmodel.builder) nonlinear_eq_from_parts() (in module jaxmodel) (in module jaxmodel.blocks) nonlinear_ineq_from_parts() (in module jaxmodel) (in module jaxmodel.blocks)

O

objective (jaxmodel.approximations.SQPSubproblemData attribute) (jaxmodel.SQPSubproblemData attribute) (kkthn.builder.KKTHardNet attribute)

objective_value() (jaxmodel.JaxNLPModel method) (jaxmodel.model.JaxNLPModel method) optimize() (kkthn.builder.KKTHardNet method)

P

pack() (jaxmodel.variables.VariableSpec method) (jaxmodel.VariableSpec method) pack_vars() (jaxmodel.JaxNLPModel method) (jaxmodel.model.JaxNLPModel method) parameter_names (kkthn.builder.KKTHardNet attribute) parameter_spec (jaxmodel.builder.NLPBuilder attribute) (jaxmodel.NLPBuilder attribute) parameters_path (kkthn.builder.DatasetSpec attribute) ParameterSpec (class in jaxmodel) (class in jaxmodel.parameters)

parse_constraint() (in module kkthn.string_problem) plot_history() (kkthn.builder.KKTHardNet method) predict() (kkthn.builder.KKTHardNet method) print_every (kkthn.training.KKTTrainConfig attribute) ProblemBuilder (in module kkthn.builder) project (kkthn.projection.ProjectionLayer attribute) project() (kkthn.native_projection.NativeProjection method) project_single (kkthn.projection.ProjectionLayer attribute) projection (kkthn.training.KKTTrainConfig attribute) ProjectionLayer (class in kkthn.projection) ProjectionSettings (class in kkthn.projection)

Q

Q (jaxmodel.approximations.QuadraticObjectiveData attribute) (jaxmodel.objective.QuadraticObjective attribute) (jaxmodel.QuadraticObjective attribute) (jaxmodel.QuadraticObjectiveData attribute) Q_diag (jaxmodel.approximations.QuadraticObjectiveData attribute) (jaxmodel.QuadraticObjectiveData attribute) quad() (kkthn.builder.KKTHardNet method) quadratic_eq_scalar() (in module jaxmodel) (in module jaxmodel.constraints) quadratic_ineq_scalar() (in module jaxmodel) (in module jaxmodel.constraints)

quadratic_objective_data() (jaxmodel.JaxNLPModel method) (jaxmodel.model.JaxNLPModel method) quadraticize_objective() (in module jaxmodel.approximations) (jaxmodel.JaxNLPModel method) (jaxmodel.model.JaxNLPModel method) quadraticize_objective_data() (in module jaxmodel.approximations) QuadraticObjective (class in jaxmodel) (class in jaxmodel.objective) QuadraticObjectiveData (class in jaxmodel) (class in jaxmodel.approximations)

R

residual (kkthn.builder.Constraint attribute)

resolve_objective() (in module jaxmodel.objective)

S

seed (kkthn.training.KKTTrainConfig attribute) set_affine_lower_bound() (jaxmodel.highlevel.HighLevelNLPBuilder method) (jaxmodel.HighLevelNLPBuilder method) set_affine_upper_bound() (jaxmodel.highlevel.HighLevelNLPBuilder method) (jaxmodel.HighLevelNLPBuilder method) set_bounds() (jaxmodel.builder.NLPBuilder method) (jaxmodel.NLPBuilder method) set_dataset() (kkthn.builder.KKTHardNet method) set_objective() (jaxmodel.builder.NLPBuilder method) (jaxmodel.highlevel.HighLevelNLPBuilder method) (jaxmodel.HighLevelNLPBuilder method) (jaxmodel.NLPBuilder method) set_projection_config() (kkthn.builder.KKTHardNet method) set_quadratic_objective() (jaxmodel.highlevel.HighLevelNLPBuilder method) (jaxmodel.HighLevelNLPBuilder method) set_train_config() (kkthn.builder.KKTHardNet method) settings (kkthn.projection.ProjectionLayer attribute) shapes (jaxmodel.parameters.ParameterSpec attribute) (jaxmodel.ParameterSpec attribute) (jaxmodel.variables.VariableSpec attribute) (jaxmodel.VariableSpec attribute)

sin() (kkthn.builder.KKTHardNet method) sizes (jaxmodel.variables.VariableSpec attribute) (jaxmodel.VariableSpec attribute) slices (jaxmodel.variables.VariableSpec attribute) (jaxmodel.VariableSpec attribute) solve_single (kkthn.projection.ProjectionLayer attribute) sqp_data() (jaxmodel.JaxNLPModel method) (jaxmodel.model.JaxNLPModel method) sqp_subproblem_data() (jaxmodel.JaxNLPModel method) (jaxmodel.model.JaxNLPModel method) SQPSubproblemData (class in jaxmodel) (class in jaxmodel.approximations) sqrt() (kkthn.builder.KKTHardNet method) structure (jaxmodel.ConstraintEntry attribute) (jaxmodel.constraints.ConstraintEntry attribute) summary() (kkthn.builder.KKTHardNet method)

T

tensor() (kkthn.builder.KKTHardNet method) text (kkthn.builder.Constraint attribute) total_size (jaxmodel.variables.VariableSpec attribute) (jaxmodel.VariableSpec attribute)

train_frac (kkthn.training.KKTTrainConfig attribute) train_kkt_hardnet() (in module kkthn.training)

U

u (jaxmodel.approximations.SQPSubproblemData attribute) (jaxmodel.SQPSubproblemData attribute) unpack() (jaxmodel.variables.VariableSpec method) (jaxmodel.VariableSpec method) unpack_vars() (jaxmodel.JaxNLPModel method) (jaxmodel.model.JaxNLPModel method)

upper_bounds() (jaxmodel.JaxNLPModel method) (jaxmodel.model.JaxNLPModel method) upper_fun (jaxmodel.bounds.BoundSpec attribute) (jaxmodel.BoundSpec attribute) use_dataset() (kkthn.builder.KKTHardNet method)

V

validate() (jaxmodel.parameters.ParameterSpec method) (jaxmodel.ParameterSpec method) validate_params() (jaxmodel.JaxNLPModel method) (jaxmodel.model.JaxNLPModel method) value_from_flat() (jaxmodel.objective.QuadraticObjective method) (jaxmodel.QuadraticObjective method) var_spec (jaxmodel.builder.NLPBuilder attribute) (jaxmodel.NLPBuilder attribute)

variable_names (kkthn.builder.KKTHardNet attribute) VariableBuilder (class in jaxmodel) (class in jaxmodel.variables) variables_path (kkthn.builder.DatasetSpec attribute) VariableSpec (class in jaxmodel) (class in jaxmodel.variables) vector() (kkthn.builder.KKTHardNet method) VectorExpression (class in kkthn.builder)

W

wrap_scalar_objective() (in module jaxmodel.objective)

wrap_vector_constraint() (in module jaxmodel.constraints)