flax
6f7322db - Use jax.nn.dot_product_attention when possible

Commit

268 days ago

Use jax.nn.dot_product_attention when possible

References

#4649 - Use jax.nn.dot_product_attention when possible

Author

IvyZX

IvyZX

Parents

Loading