Question HDRP Layered lit with custom shader

Hi,

We are trying to include the layered lit shader in our game.
From what I understand it only works with layers of materials using the standard lit shader?
Correct me if I am wrong on that

We however have a custom shader made in shadergraph based on the standard lit shader but with some extra inputs and input calculations. (mostly to apply colors to the albedo through a mask)
I was wondering if it would technicaly be possible to create a tweaked version of the layered lit shader so it would work with this custom shader (which basically is still standard lit), and if somebody could point me in the right direction of how to tweak the shader code of the layered lit shader.
I imagine we would have to add something to copy over the extra inputs and feed that into the same shader. Or have it run on our shader in the background.

Thanks,
Pieter