🐛 Fix the convergence of loss function for the llama fine tuning example by baptistecolle · Pull Request #132 · huggingface/optimum-tpu