Add self-supervised losses#1695

ahmedtaha100 wants to merge 1 commit into

google-deepmind:maingoogle-deepmind/optax:mainfrom

ahmedtaha100:feat/self-supervised-lossesahmedtaha100/optax:feat/self-supervised-lossesCopy head branch name to clipboard