[NVIDIA] Update JAX doc to include jax.nn.dot_product_attention#22450
Merged
copybara-service[bot] merged 1 commit intojax-ml:mainfrom kaixih:update_sdpa_docJul 15, 2024
+1
Commits
Commits on Jul 15, 2024
- committed
