Skip to content

[Pallas TPU] Add lowerings for bf16 jnp.ceil and jnp.floor in TPU v6+ #2436

[Pallas TPU] Add lowerings for bf16 jnp.ceil and jnp.floor in TPU v6+

[Pallas TPU] Add lowerings for bf16 jnp.ceil and jnp.floor in TPU v6+ #2436