jax.experimental.pallas.triton.approx_tanh#
- jax.experimental.pallas.triton.approx_tanh(x)[source]#
Elementwise approximate hyperbolic tangent: \(\mathrm{tanh}(x)\).
See https://docs.nvidia.com/cuda/parallel-thread-execution/index.html#floating-point-instructions-tanh.