flax
Added is_causal mask argument to flax.nnx.dot_product_attention
#5093
Merged

Loading