You are working on the TorchSim/torch-sim repository. The source code is checked out at /testbed at commit…

You are working on the TorchSim/torch-sim repository. The source code is checked out at /testbed at commit…: a task in SWE-Lego-RL-2699 (Harbor dataset). Currently TorchSim transfers the cell to numpy to use the function before transferring it back to torch.