jax

5977fc2e - Merge pull request #31131 from mattjj:discharge-state-caching