``grid_shape`` works for me.  Another option is to separate the arguments into
``rows`` & ``cols``, then you would only need to specify one, a-la
``plt.subplots``.