Denote where we should remove temporary backwards compatibility once the `weight_sum` deprecation is complete.
Denote where we should remove temporary backwards compatibility once the `weight_sum` deprecation is complete.

PiperOrigin-RevId: 440431495

google-research/t5x