I have this same problem. I have found that if, right before bed, I find some single simple puzzle or problem to think about I can mute the other stuff and just focus on that. I feel like it helps me fall asleep faster.