Skip to content

[Jax][Pallas][Mosaic] Implement platform dependent diag, with branch selection driven by constant prop in mosaic lowering. #972

[Jax][Pallas][Mosaic] Implement platform dependent diag, with branch selection driven by constant prop in mosaic lowering.

[Jax][Pallas][Mosaic] Implement platform dependent diag, with branch selection driven by constant prop in mosaic lowering. #972

Bazel single accelerator GPU tests (linux-x86-n2-16, Python 3.12, x64=1)

succeeded Dec 22, 2024 in 8m 54s