mirror of
https://github.com/ROCm/jax.git
synced 2025-04-19 05:16:06 +00:00
0b4800a193
Building on #21925, this tutorial demonstrates the use of the FFI using `ffi_call` with a simple example. I don't think this should cover all of the most advanced use cases, but it should be sufficient for the most common examples. I think it would be useful to eventually replace the existing CUDA tutorial, but I'm not sure that it'll get there in the first draft. As an added benefit, this also runs a simple test (akin to `docs/cuda_custom_call`) which actually executes using a tool chain that open source users would use in practice.
45 lines
719 B
ReStructuredText
45 lines
719 B
ReStructuredText
.. _user-guides:
|
|
|
|
User Guides
|
|
===========
|
|
|
|
User guides are deeper dives into particular topics within JAX
|
|
that become relevant as your JAX project matures into larger
|
|
or deployed codebases.
|
|
|
|
.. toctree::
|
|
:maxdepth: 1
|
|
:caption: Debugging and Performance
|
|
|
|
notebooks/thinking_in_jax
|
|
profiling
|
|
device_memory_profiling
|
|
debugging/index
|
|
gpu_performance_tips
|
|
persistent_compilation_cache
|
|
|
|
.. toctree::
|
|
:maxdepth: 1
|
|
:caption: Development
|
|
|
|
jaxpr
|
|
notebooks/external_callbacks
|
|
type_promotion
|
|
pytrees
|
|
|
|
.. toctree::
|
|
:maxdepth: 1
|
|
:caption: Run Time
|
|
|
|
aot
|
|
export/index
|
|
errors
|
|
transfer_guard
|
|
|
|
.. toctree::
|
|
:maxdepth: 1
|
|
:caption: Custom Operations
|
|
|
|
pallas/index
|
|
ffi
|