Refer more here: #12443
More context under "Result: Single coherent image matching single-GPU output" here.
Before proceeding with a PR, please reply to this issue mentioning the proposal you have in mind, and we can take it from there.
Please mention @DN6 @yiyixuxu and myself (@sayakpaul).