jaxincell._routing

Functions

merge_parameter_trees(base_parameters, override_parameters)	Recursively merge two parameter trees, with override_parameters taking precedence over base_parameters.
put_parameter_path(container, path, value)
iter_species_parameter_groups(species_type, species_values)	Iterate over nested or loose species parameter groups.
route_nested_initial_species_parameters(...)
route_flat_initial_parameters(input_parameters, ...)
build_runtime_flat_parameter_routes()
build_runtime_species_label_routes(species_parameters)
clean_runtime_input_parameters(input_parameters, ...)
build_runtime_parameter_sections(...)

Module Contents

jaxincell._routing.merge_parameter_trees(base_parameters, override_parameters)

Recursively merge two parameter trees, with override_parameters taking precedence over base_parameters.

jaxincell._routing.put_parameter_path(container, path, value)

jaxincell._routing.iter_species_parameter_groups(species_type, species_values, strict=False)

Iterate over nested or loose species parameter groups.

jaxincell._routing.route_nested_initial_species_parameters(input_parameters, parameters, differentiable_parameters, cleaner_input_parameters)

jaxincell._routing.route_flat_initial_parameters(input_parameters, parameters, differentiable_parameters, cleaner_input_parameters)

jaxincell._routing.build_runtime_flat_parameter_routes()

jaxincell._routing.build_runtime_species_label_routes(species_parameters)

jaxincell._routing.clean_runtime_input_parameters(input_parameters, runtime_flat_parameter_routes, runtime_species_label_routes, species_parameters)

jaxincell._routing.build_runtime_parameter_sections(base_parameter_sections, input_parameters)