We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
773cdfe
lax.select generates a nan in the gradient from the inactive branch, while lax.cond doesn't
lax.select
lax.cond