AI/ML Software Engineer

Job Family : Software Development & Support Travel Required : None Clearance Required : Ability to Obtain Public Trust What You Will Do Keystone AI Technology is seeking a highly skilled and motivated AI-focused Software Engineer to join our dynamic team. In this role, you will work with federal clients to design, develop, and deploy innovative Artificial Intelligence and Machine Learning solutions that address immediate mission challenges. You will be a key contributor on a team of data scientists, developers, and subject matter experts, applying best practices to build robust AI applications, sophisticated data pipelines, and intelligent systems. The ideal candidate is a fast learner and rapid developer with a strong foundation in software engineering and demonstrable experience building production-ready AI solutions from concept to deployment. The ideal candidate should also possess strong hands-on expertise in Python, machine learning, graph analytics, natural language processing, and entity resolution. This combination of skills is essential for uncovering hidden patterns across large, disparate datasets in support of fraud detection, trafficking detection, and other mission-critical investigative use cases. A critical competency for this role is network analysis—the ability to link individuals, transactions, organizations, events, documents, and locations to identify relationships, detect anomalies, and expose organized illegal activities. In tandem, the candidate should be proficient in developing and applying AI/ML models for risk scoring, anomaly detection, pattern recognition, and investigative prioritization, enabling federal clients to enhance operational effectiveness and focus resources on the highest-risk cases. What You Will Do Design, develop, and deploy end-to-end AI/ML applications, with a focus on Retrieval-Augmented Generation systems, AI chatbots, and agentic workflows. Construct and maintain scalable data pipelines for processing, transforming, linking, and feeding structured and unstructured data into AI models and applications. Develop machine learning models for risk scoring, anomaly detection, classification, clustering, and predictive analytics to support investigative and operational decision-making. Apply graph analytics and network analysis techniques to identify hidden relationships among individuals, transactions, locations, organizations, and events. Implement entity resolution capabilities to match, deduplicate, and link records across large, disparate datasets. Apply NLP techniques to extract entities, relationships, topics, events, and indicators from unstructured text, documents, case notes, and other mission-relevant data sources. Collaborate with cross-functional teams to rapidly prototype and iterate on solutions for mission-critical challenges in areas such as fraud detection, trafficking detection, data triage, search and optimization, and automated discovery. Develop and maintain REST APIs to serve model inferences and integrate AI capabilities into larger systems. Contribute to the team's software architecture design, ensuring solutions are scalable, reliable, secure, and efficient. Deploy and manage applications and models in cloud environments, primarily AWS, leveraging infrastructure-as-code and DevOps best practices. What You Will Need Must be able to OBTAIN and MAINTAIN a Federal or DoD "PUBLIC TRUST"; candidates must obtain approved adjudication of their PUBLIC TRUST prior to onboarding with Keystone AI . Candidates with an ACTIVE PUBLIC TRUST or SUITABILITY are preferred. Bachelor’s degree is required. Minimum SIX (6) years of hands-on experience with Python and JavaScript and/or TypeScript. A Master’s degree OR Industry Certification may be substituted for up to two years of relevant professional experience. Strong hands-on experience with Python for AI/ML application development, data engineering, data analysis, and model implementation. Proven experience developing RAG applications, including implementing re-ranking strategies. Experience building AI chatbots or conversational agents. Hands-on experience with AI application frameworks such as LangChain, Haystack, crewAI, or similar. Strong knowledge of core Python data science and ML libraries, including NumPy, Pandas, Scikit-learn, NLTK, and OpenCV. Demonstrated experience developing and applying machine learning models for classification, clustering, risk scoring, anomaly detection, and pattern recognition. Experience with graph analytics, network analysis, and relationship discovery across complex datasets. Experience with entity resolution, record linkage, deduplication, identity matching, or similar data matching techniques. Experience applying NLP techniques such as named entity recognition, text classification, semantic search, information extraction, topic modeling, and relationship extraction. Familiarity with deep learning frameworks such as PyTorch or TensorFlow. Experience with search technolo