{"library":"accelerate","install":[{"cmd":"pip install accelerate","imports":["from accelerate import Accelerator\n\ndef training_function():\n    # Accelerator MUST be initialized inside the training function for notebook_launcher\n    accelerator = Accelerator(mixed_precision='fp16')\n    model, optimizer, dataloader = accelerator.prepare(model, optimizer, dataloader)\n    \n    for batch in dataloader:\n        optimizer.zero_grad()\n        loss = model(batch)\n        accelerator.backward(loss)  # NOT loss.backward()\n        optimizer.step()","loss = criterion(outputs, targets)\naccelerator.backward(loss)"]},{"cmd":"accelerate config","imports":[]},{"cmd":"python -c \"from accelerate.utils import write_basic_config; write_basic_config(mixed_precision='fp16')\"","imports":[]}]}