Replace external rlax vtrace dependency after function arg has been renamed.
Replace external rlax vtrace dependency after function arg has been renamed.

PiperOrigin-RevId: 353211960
Change-Id: Ifdfef316fbf0425b6f8186c3b6208da9708dc3f8

deepmind/acme,deepmind/acme