src/transformers/models/clip/modeling_flax_clip.py/FlaxCLIPMLP/__call__
class FlaxCLIPMLP:    def __call__(self, hidden_states):
        hidden_states = self.fc1(hidden_states)
        hidden_states = self.activation_fn(hidden_states)
        hidden_states = self.fc2(hidden_states)
        return hidden_states