Chip-scale atomic wave-meter enabled by machine learning