Deep learning for Lagrangian drift simulation at the sea surface