AI-assisted quantum sensing