flax
Use jax.nn.dot_product_attention when possible
#4649
Merged

Commits
  • Use jax.nn.dot_product_attention when possible
    IvyZX committed 274 days ago
Loading