[Pallas TPU] Add WeirdOp
to TPU dialect and add lowering for lax.is_finite
#39
Loading
WeirdOp
to TPU dialect and add lowering for lax.is_finite
#39