Skip to content

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

[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. #969

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

succeeded Dec 22, 2024 in 7m 11s