Most recent commit, I added support for using most of the functions using jax. I was hoping to kill both the lazy backends, but for whatever reason I can't get dask to work.

I also have added some of the backends to the core tests.

The only one I can't quite get to work is the cylinder segment, partially just because it's a lot of code, and the functions for each case are (n,) -> (n, 3, 3) rather than (n,) -> (n,)
I'm sure there's a way to make it work without a huge refactor, but I couldn't make it work instantly. I'll mess with it a bit more and come up with what I think is the best solution here, and whether this needs a refactor or not.