Replace torch.jit.script with torch.compile (PyTorch 2.x migration)#299

Copilot wants to merge 2 commits intomainAdaptiveMotorControlLab/CEBRA:mainfrom

copilot/update-torch-jit-script-usageAdaptiveMotorControlLab/CEBRA:copilot/update-torch-jit-script-usageCopy head branch name to clipboard