Foundation models for atoms