jax.lax.scatter\_add ==================== .. currentmodule:: jax.lax .. autofunction:: scatter_add