Stock ROM with ambient display?

I'm using CM 12.1, I love it but I'm having a lot of FCs and reboots that I didn't used to have with stock. Also, I love on stock the double twist inside the camera app which flips between front camera and rear camera. The only thing that makes me keep using CM is the ambient display. Is there any custom rom based on stock that does have ambient display working with the IR and pick up sensors?