Added is_causal mask argument to flax.nnx.dot_product_attention#5093
Merged
copybara-service[bot] merged 1 commit intogoogle:mainfrom Dec 11, 2025
Merged
Added is_causal mask argument to flax.nnx.dot_product_attention#5093copybara-service[bot] merged 1 commit intogoogle:mainfrom
copybara-service[bot] merged 1 commit intogoogle:mainfrom
