The memory_efficient=True option uses internally torch.utils.checkpoint to trade compute for memory.
This operator is not defined in ONNX, which raises this error.
However, if you don’t necessarily need checkpointing, you could just set this argument to False and the export should work.