We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
jax.ops.index_update(x, idx, y)
x.at[idx].set(y)