[JAX] Adds computation backend information to serialized Meridian models.#1444

copybara-service[bot] merged 1 commit intomaingoogle/meridian:mainfrom

test_863396043google/meridian:test_863396043Copy head branch name to clipboard