jax.experimental.pallas.triton
module#
Triton-specific Pallas APIs.
Classes#
|
Compiler parameters for Triton. |
Functions#
|
Elementwise approximate hyperbolic tangent: \(\mathrm{tanh}(x)\). |
Synchronizes all kernel executions in the grid. |
|
|
Inline assembly applying an elementwise operation. |