There are a few apps that can do this on a per app basis like wakey but they require a purchase.

Personally I just use Waze and the yt music integration, waze on its own keeps the screen on

There also xposed modules but that requires a rooted phone.