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