Machine learning-enabled battery design