We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
jax.nn.one_hot(x, num_classes, dtype)
broadcast_argnums
pmap
core.call_p