Since relu is used as a functional call, the cleanest way would be to override the class and reimplement the forward method with your new non-linearity.
By doing so, you could also create the new activation function as an nn.Module, so that it can be replaced easily, if that’s your use case.