I haven't tried it myself but niri might do what you want using Wayland https://github.com/niri-wm/niri