Allow for log scaling of media

[romin-ebrahimi]

In some cases, the performance of these models can be greatly improved with the use of natural log scaling. The current scaling options are restricted to commutative operations like division and multiplication. It would be helpful if the preprocessing module added the functionality to apply log scaling.

[romin-ebrahimi]

What does this have to do with log scaling?

How to perform non zero scaling? ""lambda x: jnp.mean(x[x > 0])"" is not working. Thanks in advance!