jax

07458292 - [Mosaic GPU] Explicitly set kernel_name on both Pallas and plain Mosaic GPU codepaths.