src/transformers/models/gptj/modeling_flax_gptj.py/FlaxGPTJMLP/__call__
class FlaxGPTJMLP:    def __call__(self, hidden_states, deterministic: bool = True):
        hidden_states = self.fc_in(hidden_states)
        hidden_states = self.act(hidden_states)
        hidden_states = self.fc_out(hidden_states)
        hidden_states = self.dropout(hidden_states, deterministic=deterministic)
        return hidden_states