Add support for transformations in GaussianMixture.
Add support for transformations in GaussianMixture.

PiperOrigin-RevId: 469987113
Change-Id: I5d4558ab2bd25337e55063ffe361412e56593d81

deepmind/acme,deepmind/acme