b0/configurations/b0_ax_joblib.yml

0001 problem:
0002   name: "ePIC B0 Geometry Optimization"
0003   problem_type: "EPIC_B0"
0004
0005   output_location: "./output/ax_joblib"
0006   work_location: "./work/ax_joblib"
0007   epic_environment:
0008     eic_shell: "/home/cua/eic/eic-shell"
0009     eic_shell: "/path/to/my/eic-shell"
0010     epic_install: "./epic"
0011     epic_config: "epic"
0012     geometry_mode: "build"
0013
0014   inline_design:
0015     epic_design_space:
0016       epic_design_parameters:
0017
0018         b0_tracker:
0019           file_path: "$EPIC_INSTALL/compact/far_forward/B0_tracker.xml"
0020
0021           parameters:
0022
0023             z1:
0024               value: -50.0
0025               bounds: [-50.0, -30.0]
0026               xml_path: ".//constant[@name='B0TrackerLayer1_zstart']"
0027               attribute: "value"
0028               unit: "cm"
0029
0030             z2:
0031               value: -20.0
0032               bounds: [-20.0, -5.0]
0033               xml_path: ".//constant[@name='B0TrackerLayer2_zstart']"
0034               attribute: "value"
0035               unit: "cm"
0036
0037             z3:
0038               value: 0.0
0039               bounds: [0.0, 20.0]
0040               xml_path: ".//constant[@name='B0TrackerLayer3_zstart']"
0041               attribute: "value"
0042               unit: "cm"
0043
0044             z4:
0045               value: 30.0
0046               bounds: [30.0, 40.0]
0047               xml_path: ".//constant[@name='B0TrackerLayer4_zstart']"
0048               attribute: "value"
0049               unit: "cm"
0050
0051       optimization_groups:
0052         default:
0053           - "b0_tracker.z1"
0054           - "b0_tracker.z2"
0055           - "b0_tracker.z3"
0056           - "b0_tracker.z4"
0057
0058   objectives:
0059     - name: "lever_arm"
0060       direction: "maximize"
0061
0062 optimizer:
0063   name: "ax"
0064   type: "bayesian"
0065   parameters:
0066     initialization_strategy: "sobol"
0067     generator: "BOTORCH_MODULAR"
0068     n_initial_samples: 1
0069     n_iterations: 2
0070     batch_size: 1
0071     seed: 42
0072
0073 scheduler:
0074   runner_type: "JobLibRunner"
0075   parameters:
0076     n_jobs: 1
0077     backend: "threading"
0078
0079 workflows:
0080   workflows:
0081     - name: "b0_geometry_validation"
0082       description: "Validate B0 geometry modification with a placeholder objective"
0083       stack_type: "epic"
0084
0085       branches:
0086         - name: "main"
0087
0088           stages:
0089
0090             - name: "geo"
0091               description: "Prepare and validate modified B0 geometry"
0092
0093               jobs:
0094                 - name: "geo_job"
0095
0096                   payload:
0097                     evaluator_type: "stack"
0098                     stack_type: "epic"
0099                     job_id: "geo"
0100
0101                   layers:
0102                     - name: "geo"
0103                       layer: "geo"
0104                       inputs:
0105                         - "{{geometry_dir}}/install/share/epic/epic.xml"
0106                       outputs:
0107                         - "{{output_dir}}/geo.overlaps.txt"
0108
0109             - name: "evaluate"
0110               objective_plan:
0111                 steps:
0112                   stages:
0113                     - name: "evaluate_b0"
0114                       inline:
0115                         entrypoint: "examples.epic.b0.b0_evaluator:objective_payload"
0116                       produces_objective: true