jax
eada2988 - [Mosaic] Add explicit control over core parallelization strategy

Commit
1 year ago
[Mosaic] Add explicit control over core parallelization strategy This CL introduces the dimension semantic `core_parallel`. It allows the user to control which dimension is parallelized across cores as opposed to leaving it to Mosaic to choose the best out of `parallel` dimensions. PiperOrigin-RevId: 756076809
Author
Parents
Loading