Added is_causal mask argument to flax.nnx.dot_product_attention#5093
Open
ibbyml wants to merge 1 commit intogoogle:mainfrom
Open
Added is_causal mask argument to flax.nnx.dot_product_attention#5093ibbyml wants to merge 1 commit intogoogle:mainfrom
ibbyml wants to merge 1 commit intogoogle:mainfrom
