feat: static shape jaten ops for multimodal embedding#96

kyuyeunk merged 2 commits into

google:maingoogle/torchax:mainfrom

anthonsu:aysu/gemma4-recompileanthonsu/torchax:aysu/gemma4-recompileCopy head branch name to clipboard