Remove converting list_args_to_tuple for caching as mesh_axes is now a Pspec
Remove converting list_args_to_tuple for caching as mesh_axes is now a Pspec

PiperOrigin-RevId: 449901578

google/jax,google/jax,google/jax,google/jax