Adds `distribution()` convenience function.

I'm tired of typing `Distribution.from_shape` all the time.  This seems
to be the main entry point, so `distribution` is a shortcut.