def make_resid_delta_mlp_from_mlp_post_act_scalar_deriver(
    dst_config: DstConfig,
) -> ScalarDeriver:
    """Returns a scalar deriver for the write vector for the MLP layer at each token."""
    model_context = dst_config.get_model_context()
    layer_indices = dst_config.layer_indices or list(range(model_context.n_layers))

    scalar_deriver = get_base_mlp_scalar_deriver(
        dst_config=dst_config,
    )
    W_out_by_layer_index = get_mlp_write_tensor_by_layer_index(
        model_context=model_context, layer_indices=layer_indices
    )
    return convert_scalar_deriver_to_write(
        scalar_deriver=scalar_deriver,
        write_tensor_by_layer_index=W_out_by_layer_index,
        output_dst=DerivedScalarType.RESID_DELTA_MLP_FROM_MLP_POST_ACT,
    )
Returns a scalar deriver for the write vector for the MLP layer at each token.
neuron_explainer/activations/derived_scalars/mlp.py