Scientific Problem solving and Wu Wei

I am a mathematician, and I often find myself obsessing over the problems that I'm trying to solve. While I have made incredible progress over the last 6 months with my practice. I find myself suffering, sometimes for a few days in a row, because of my obsessive thinking towards the problem. It's as if my entire consciousness became the problem and that I could not find any relief until it was fixed. I enjoy getting consumed by what I do, I also know that being consumed AND remaining grounded is much more powerful. Any tip? / similar experiences?