jax.experimental.pallas.triton module

.. automodule:: jax.experimental.pallas.triton

Classes

.. autosummary:: :toctree: _autosummary

CompilerParams

Functions

.. autosummary:: :toctree: _autosummary

atomic_and atomic_add atomic_cas atomic_max atomic_min atomic_or atomic_xchg atomic_xor approx_tanh debug_barrier elementwise_inline_asm load max_contiguous store