Use `power` instead of `pow` for element-wise operator

[Rémi Louf]

This is somewhat tricky as the name power is already used for the operation on scalars.

https://numpy.org/doc/stable/reference/generated/numpy.power.html#numpy.power