jax

ce4dae1c - Add logging the jax2tf `mlir_module_serialized` module size.

Add logging the jax2tf `mlir_module_serialized` module size. PiperOrigin-RevId: 662574156 diag_docstring_added diag_docstring_desc_fixed white_space_fixed white_space_fixed white_space_fixed linting fixed