Add function that create JaxArrays

[Ericgig]

Add dispatched function that create JaxArray from parameters: zeros, identity, diags and one_elements.

[coveralls]

Pull Request Test Coverage Report for Build 3251317453

• 44 of 50 (88.0%) changed or added relevant lines in 1 file are covered.
• No unchanged relevant lines lost coverage.
• Overall coverage increased (+4.8%) to 93.897%

---

Changes Missing Coverage	Covered Lines	Changed/Added Lines	%
src/qutip_jax/create.py	44	50	88.0%
<!--	Total:	44	50	88.0%	-->

Totals
Change from base Build 3160050711:	4.8%
Covered Lines:	200
Relevant Lines:	213

---

💛 - Coveralls

[quantshah]

I made the small changes in the docstrings and pushed. I hope that is okay @Ericgig ?