It takes time and a lot of negotiation to arrive at a routine that works.
I think a big problem is that women are more accepting of the massive increase in household work load, and often make the mistake of assuming more of it to avoid the tricky negotiation (or argument). Don’t do that.
Keep dialogue open and talk about how things are going to work. There doesn’t need to be a massive change when you go back to work, you are already working hard -start getting it right now to make the transition easier for you.
First, he’s going to have to realise that it’s hard work being with a baby all day and you frequently do need to hand them over for a rest/sit down with a cup of tea when your partner gets home, before getting stuck into the dinner, bath, bed routine. If he can’t accept this through discussion, leave him with baby for a few days while you do some KIT days or just meeting people from work or something, get him to take annual leave. He’ll see how much you need to pass baby over for a bit then.
Next you need to talk about how you are both going to protect adult rest time in the evening. In our house this means he does bath time and gets the kids ready for bed while I do a run around the house -load dishwasher, washing machine, fold some clothes, make the living room a tidy place to relax in. Once the youngest is in bed, the older two go to their rooms to read, and we relax. You need this, so aim to have a little bit of housework done every day in that time, so the evening is yours.
If the dinner bed bath routine/tidying time every single day is too relentless for both of you, prevents a hobby or something, you can agree a night off for each of you. E.g. he does it all on a Tuesday, while you do it all on a Wednesday, or just alternate weeks -taking turns to have a Tuesday off. The free person can do what they like -gym, watch tv, play computer games, whatever....but the point is it;s equal. So if he has a gym habit, running club, pub night etc...it’s agreed when your time is, no matter what it’s for.
It takes time. My own DH responded to the tiredness (and let’s face it, all parents are really fucking tired all the time) by dragging himself through everything slowly, so we’d be still loading the dishwasher at 11pm. It took him a while to appreciate that meal planning and batch cooking so we could eat with the kids, and having a busy 90 minutes was worth it to have the evening back. But we’re there now.