Make JAX linearly_decaying_epsilon function gin configurable.
Make JAX linearly_decaying_epsilon function gin configurable.

PiperOrigin-RevId: 334799213

google/dopamine,google/dopamine,google/dopamine