Just sharing that there is a react component I have used called "React Wrap Balancer" from Vercel that does something similar and works today:

https://vercel.com/blog/react-wrap-balancer

It does add another dependency which I am not fond of.