Software Engineer - AI/ML Systems (Context Services)

What Keystone AI is: Relentless to achieve Keystone AI operates at the forefront of industrial digitalization, building AI , and data solutions that solve the world’s hardest, highest-impact problems. With unmatched industrial heritage and a comprehensive suite of AI capabilities, including low-code AI agents, Keystone AI accelerates the digital transformation to drive operational improvements. We thrive in challenges. We challenge assumptions. We execute with speed and ownership. If you view obstacles as signals to step forward - not backwards - you’ll feel right at home here. Our Moonshot is bold: Unlock $100B in customer value by 2035, and redefine how global industry works. Join us in this venture where AI and data meet ingenuity, and together, we will forge the path to a smarter, more connected industrial future. About the Role We are seeking a Software Engineer to join the Context Services team. You will work day-to-day with our Applied Scientists and AI Engineers to transition predictive models and LLM agents from the experimentation phase into scalable production environments. Your core focus will be on writing clean, efficient code to serve these AI solutions to our users quickly and reliably" The Impact you bring to Keystone AI Key Responsibilities: High-Performance Deployment: Package AI/ML models and LLM workflows, developing low-latency APIs to support thousands of concurrent users. Industrial Problem Solving: Partner with our AI and product teams to translate complex industrial data challenges and agentic workflows into reliable software services. Pipeline Execution: Build and maintain automated pipelines for data processing, model deployment, agent monitoring, and system evaluation. System Health: Implement monitoring solutions to track system performance, inference costs, and latency bottlenecks in production. Continuous Learning: Stay up-to-date with emerging engineering trends in ML and Generative AI, participating in research and prototyping to improve our internal systems. Qualifications & Skills: Experience: 3-6 years in Software Engineering, Backend Engineering, or DevOps. Programming: Strong Python skills and experience with modern web frameworks (such as FastAPI) and handling asynchronous tasks. Systems: Experience building containers and a solid understanding of basic CI/CD pipelines. Collaboration: A collaborative, curious mindset with a strong desire to learn the AI/LLM lifecycle and work closely with domain experts. Learn more about us Impact 2025 Keystone AI's Industrial AI: Moonshot We’re globally recognized domain experts with an international presence that spans Phoenix, Houston, Oslo Tokyo, Bengaluru, and Abu Dhabi. Equal Opportunity Keystone AI is committed to creating a diverse and inclusive environment at work and is proud to be an equal opportunity employer. All qualified applicants will receive the same level of consideration for employment.