SURROGATE MODEL BASED HEAT SINK DESIGN FOR ENERGY X2026