Investigate code speed

Investigate code speed by tweaking the following

• jnp.array(np.array(...)) vs jnp.array(...) in Quantity -- Almost same time taken by both cases.
• Type checking inside loops, e.g. compositeHamiltonian _gradients_one_time