Function absolute

Calculate the absolute value element-wise.

This is the same function as jax.numpy.abs().