as_scaled_array and asarray factory methods compatible with JAX PyTrees.

as_scaled_array and asarray factory methods compatible with JAX PyTrees. #116