WIP: Slightly more efficient stockwell

Moving toward adding `cuda` support for the Stockwell transform.

So far all I've done is gone through and made some simplifications that should make it a bit more efficient.