Add permutation specialisations

[Ericgig]

Add array and tensor permutation methods.

There is no compiling these, input accept both array, list and None. Pretty much the worth case for jit.

@quantshah This complete the dispatched functions for now. It's more complete than the Dense layer, there should only few operations that convert to other types and makes us lose the trace of jax arrays. They will mostly be operations that we have not updated to use v5's data layer yet. They are all ready to review and merge, but there will be conflict in __init__.py

[coveralls]

Pull Request Test Coverage Report for Build 3251602437

• 42 of 48 (87.5%) changed or added relevant lines in 2 files are covered.
• No unchanged relevant lines lost coverage.
• Overall coverage decreased (-1.0%) to 95.455%

---

Changes Missing Coverage	Covered Lines	Changed/Added Lines	%
src/qutip_jax/permute.py	41	47	87.23%
<!--	Total:	42	48	87.5%	-->

Totals
Change from base Build 3251591510:	-1.0%
Covered Lines:	420
Relevant Lines:	440

---

💛 - Coveralls

[quantshah]

I think you can merge this @Ericgig. There are small cosmetic changes here and there but lets go ahead and fix them later when we build docs.