Dynamics-aware Embeddings