jax

6c66e938 - [Mosaic][SC] Expose control over the number of active SC cores

[Mosaic][SC] Expose control over the number of active SC cores On SparseCore, the `core_parallel` dimension semantic allows the user to set the number of active cores to `all` or `1` based on the dimension size. PiperOrigin-RevId: 756388100