Mac has horizontal workspace switching, which is great. However, I find it very useful to have a 2D workspace grid (which I am used to from Linux) and I can switch both horizontally and vertically. Is there a way to do the same in Mac?
Not really. The Apple window managers work side to side for space switching on both iOS and macOS. Same with the split screen control - only side to side movement is programmed.
You could get an alternative window manager, but you can probably get 90% of the benefit learning the key shortcuts to move quickly to a set screen and disabling the "reorder spaces" option in Mission control.
I get why that's there for new users, but power users almost never want their spaces to be "reorganized" on the fly.