Bug Description

Doing something like

a = wp.tile_zeros(shape=(8,), dtype=float)
a[j] = x[j]

in a kernel is not differentiable. It would be nice to support tile element assignment differentiability.

System Information

No response