jax

ae80a5dc - Merge pull request #30034 from mwhittaker:heartbeat_in_initialize