Add `callback` parameter to `embed_kernel` function

Can be used by caller to get connection info.