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

Fix minor typo in Pallas docs.

Browse Source 
PiperOrigin-RevId: 625117045
This commit is contained in:
Justin Fu 2024-04-15 16:19:53 -07:00 committed by jax authors
parent cd9dcd20dd
commit f8919a32e0
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/pallas/tpu/details.rst
View File

@ -299,7 +299,7 @@ expensive (🔴).
Many JAX functions are implemented in terms of other JAX primitives, so this
list might not be comprehensive. For example, ``jax.nn.relu`` is implemented
in terms of comparisons and ``jnp.where`` and will work in Pallas kernels too.
in terms of comparisons and ``jnp.where`` will work in Pallas kernels too.
Array constructors
^^^^^^^^^^^^^^^^^^