Learning to levitate