src/transformers/generation_flax_logits_process.py/FlaxTemperatureLogitsWarper/__call__
class FlaxTemperatureLogitsWarper:    def __call__(self, input_ids: jnp.ndarray, scores: jnp.ndarray, cur_len: int) -> jnp.ndarray:
        scores = scores / self.temperature
        return scores