We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
lax.cond_p
ValueError
axis1==axis2
jnp.trace