Consider not relying on constrained layout

Overall this looks fun.

Super happy you are using constrained layout, but suggest you make it optional somehow. There are some times when it'll fail and users will want a way to turn it off.

I'm also not sure about initializing the figure inside of your helper, instead of letting the user pass it (optionally?).

Start by locating the helper that initializes the figure and enables constrained layout. Define the API needed to make constrained layout optional and to let callers provide a figure, then verify that both paths preserve the existing subplot arrangement behavior.

Written by the indexing model from the issue text.