[UPDATE JAX API] Update `jax.experimental.array_serialization.serialization` to latest JAX

[Steboss]

In the latest JAX version there's been a modification of the behaviour of jax.experiment.array_serialization.serialization where the attributes:

• _spec_has_metadata
• ts have been transferred to ts_impl (here). Thus I modified the array_serialization.py code that uses serialization via ts

@matthew-e-hopkins for visibility

[github-actions[bot]]

This pull request has been automatically marked as stale because it has been inactive for 60 days. It will be closed in 7 days if no further activity occurs. If you would like to continue working on this, please remove the stale label or leave a comment.