Not sure why the person below with the office rental was downvoted, but it's a reasonable compromise. Having a space to go to can be really important, as parenting can really eat up headspace.

My first contract/WFH job overlapped with when my first child was born and it was hell and I ended up just packing it in and returning to full-time in office work. Just too distracting.

I've been working from home for even longer and working from home when my children were born was awesome. Being able to be a bigger part of the first year for both children was great for our whole family. Having the flexibility to do so was great. I definitely adjusted how and when I worked but I wouldn't have traded that for anything.

As someone a few years ahead of you, what you may want the most is flexibility - the flexibility to decide to work from home when you want, but the option to hot desk in an office when you just need a break or distance from hectic home life. The most attractive employers may be the ones who strike this balance.

I was remote pre-pandemic and it's much harder with a toddler than it was without a toddler.