dumanah

Results 1 comments of dumanah

> You could do this with `lax.cond` inside `lax.scan` -- just apply the identity function if the loop should terminate early. > > The downside is that you will pay...