Generalize dice_loss with alpha/beta weighting#1458

copybara-service[bot] merged 1 commit intogoogle-deepmind:maingoogle-deepmind/optax:mainfrom

aymuos15:dice_ceaymuos15/optax:dice_ceCopy head branch name to clipboard