mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 12:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update quickstart.md

Browse Source 
fix minor grammar typo
This commit is contained in:
bion howard 2024-07-09 12:48:51 -04:00 committed by GitHub
parent 0dff794f68
commit 1ace88bfba
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/quickstart.md
View File

@ -158,7 +158,7 @@ Another useful transformation is {func}`~jax.vmap`, the vectorizing map.
It has the familiar semantics of mapping a function along array axes, but instead of explicitly looping
over function calls, it transforms the function into a natively vectorized version for better performance.
When composed with {func}`~jax.jit`, it can be just as performant as manually rewriting your function
operate over an extra batch dimension.
to operate over an extra batch dimension.
We're going to work with a simple example, and promote matrix-vector products into matrix-matrix products using {func}`~jax.vmap`.
Although this is easy to do by hand in this specific case, the same technique can apply to more complicated functions.