🚀 Feature

Add torch.nn.GELU as the module for GELU activation.

Motivation

Currently we have a torch.nn.functional.gelu op but no corresponding module like we do for the other activations (e.g. nn.ReLU()).

Pitch

Alternatives

Additional context