Epsilon keeps getting reintialized in the example script

In [simple_dqn.py](https://github.com/deepmind/rlax/blob/master/examples/simple_dqn.py), the epsilon schedule is chosen by actor_state.count however, that keeps getting reinitialized after every episode in [experiment.py](https://github.com/deepmind/rlax/blob/master/examples/experiment.py). So the epsilon schedule does not work.