Use zeros as initialization for avg_sq_grad in rmsprop
Use zeros as initialization for avg_sq_grad in rmsprop

This seems more sensitive thing to do than ones. It is also consistent with Keras, PyTorch and Sonnet 2's implementation.
google/jax,google/jax,tensorflow/probability,google/jax,google/jax,tensorflow/probability