Multi-fidelity learning for atomistic models via trainable data embeddings