That thing you’ve linked would probably work, but it’s an expensive one designed for an iPad Pro. These sort of things are available easily and cheaply and work fine. I paid about $10 for a three-pack of a different brand.
Hang on a tic - how are you planning to use the grid with Ableton? There’s no direct support for it inside Ableton itself (by which I mean: you are not going to use it to launch clips, nor as a MIDI controller, because grids don’t speak MIDI).
The documentation page has full details on setup: you need to install serialosc, and then other software that supports grids will be able to communicate with your hardware. A common way to use a grid directly with Ableton is via one of the Max4Live patches that support it (several are linked on that documentation page). M4L is part of Live Suite edition, or can be purchased separately.
The grid can also talk directly to standalone Max patches, and you can run these on a trial version of Max just fine - but you won’t be able to save/alter the patches.