Implement moving averages in terms of the public API.
Implement moving averages in terms of the public API.

PiperOrigin-RevId: 339215689
Change-Id: Ied25f7601d4908757b6b9bf247d923d147082a61

deepmind/dm-haiku,deepmind/dm-haiku