DSV4: Add mask hoisting, boolean splash masks, and thresholded indexer membership - #4914

DSV4: Add mask hoisting, boolean splash masks, and thresholded indexer membership#4914

systalyze-ai wants to merge 2 commits into

AI-Hypercomputer:mainAI-Hypercomputer/maxtext:mainfrom

systalyze-ai:upstream-pr/hoisted-bool-attention-maskssystalyze-ai/maxtext:upstream-pr/hoisted-bool-attention-masksCopy head branch name to clipboard