flax
9bab1485 - Add is_causal mask argument and tests

Commit

64 days ago

Add is_causal mask argument and tests

References

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

Author

ibbyml

ibbyml

Parents

Loading