[BC] Add support for optimizers that require params
[BC] Add support for optimizers that require params

PiperOrigin-RevId: 383584528
Change-Id: I5693edd8456d2f2d1b62deea40f5f221a93615c0

deepmind/acme,deepmind/acme