Switch to approximate EMA PCEN.
Switch to approximate EMA PCEN.

This change bypasses a lax.scan bug in jax2tf. The long window size (256) provides a very good approximation of the full EMA.

PiperOrigin-RevId: 490752574

google-research/chirp