This document is relevant for: Trn1, Trn2, Trn3

nki.language.silu_dx#

nki.language.silu_dx(x, dtype=None)[source]#

Derivative of SiLU activation, element-wise.

This document is relevant for: Trn1, Trn2, Trn3