This document is relevant for: Trn1, Trn2, Trn3 nki.language.gelu_dx# nki.language.gelu_dx(x, dtype=None)[source]# Derivative of GELU activation, element-wise. This document is relevant for: Trn1, Trn2, Trn3