Update jax doc sdpa

2025-04-14 10:56:06 +00:00 · 2024-07-15 17:30:54 +00:00 · 2024-07-15 17:30:54 +00:00 · 0d387e0839
commit 0d387e0839
parent 26ec43f9e5
1 changed files with 1 additions and 0 deletions
--- a/docs/jax.nn.rst
+++ b/docs/jax.nn.rst
@ -52,3 +52,4 @@ Other functions
    logsumexp
    standardize
    one_hot
+    dot_product_attention