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

[pallas] Add a warning message about experimental and incomplete status

Browse Source
This commit is contained in:
George Necula 2024-08-07 08:38:56 +03:00
parent dd958adc39
commit 64eb8e9639
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
docs/pallas/index.rst
View File

@ -3,6 +3,9 @@
Pallas: a JAX kernel language
=============================
Pallas is an extension to JAX that enables writing custom kernels for GPU and TPU.
It aims to provide fine-grained control over the generated code, combined with
the high-level ergonomics of JAX tracing and the `jax.numpy` API.
This section contains tutorials, guides and examples for using Pallas.
See also the :class:`jax.experimental.pallas` module API documentation.
@ -10,6 +13,10 @@ See also the :class:`jax.experimental.pallas` module API documentation.
Pallas is experimental and is changing frequently.
See the :ref:`pallas-changelog` for the recent changes.
You can expect to encounter errors and unimplemented cases, e.g., when
lowering of high-level JAX concepts that would require emulation,
or simply because Pallas is still under development.
.. toctree::
:caption: Guides
:maxdepth: 2