This is one way of doing it (without the squishing on the top and bottom, or do you want that?):
There are 3 things not visible directly on the screenshot:
- Right click on the surface parameter and set it to reparameterize.
- The x and y inputs of the scale component has the expression 1/x and 1/y respectively.
- The input of the bounding box component is set to flatten.