If I were running a company, I’d try and offer generous parental leave with the option of weaning out of / back into full time work. Maybe a few weeks of part-time work, a month or two off, and then back to part-time work (as desired) to help adjust.