jax

3ea39edf - Eliminate skiptest for C API in `memories_test` and add TPU C API config

Eliminate skiptest for C API in `memories_test` and add TPU C API config GPU [already skips](https://github.com/jax-ml/jax/blob/5c76ba1e56bcff30b6b47b998f5b6727f44b3ea6/tests/memories_test.py#L714) this class of test. PiperOrigin-RevId: 783075225