better installation organization

[kushalkolar]

Hi I know this is a WIP but posting here so they're not forgotten, these are w.r.t. the develop branch:

• requires dredge to be installed, I installed it directly from the repo and it seems fine (can run the nb you gave last week)
• requires seaborn to import even though it's in optional dependencies. If you think it should be a required dep I'd move it to a required dep, otherwise segregate the code that uses seaborn from always eager importing
• jaxtyping missing from deps

[cwindolf]

This should be worked out on main/develop now, thanks!