jax

3904ced2 - [Mosaic] Test only cl - add triu test, skip bf16 due to select being native bitwidth only