CoreAI Principal Engineer

Keystone.ai is a global technology and services firm specializing in enterprise AI, economics advisory and technology strategy. Keystone’s Deep Enterprise™ AI platform—built by AI/ML industry pioneers—helps large enterprises optimize business decisions at scale. By combining advanced technology, strategic consulting, and applied econometrics, Keystone delivers innovative solutions to organizations across the technology, business, legal, and government sectors. Founded in 2003, Keystone operates globally with offices in New York, San Francisco, Boston, Seattle, London, and Dubai.

Keystone’s CoreAI Group employs the world’s best AI/ML science and technology practitioners with unparalleled experience implementing highly complex, massive-scale algorithms and models to help companies make better decisions across manufacturing, supply chain management, sales, and marketing.

We are based Bellevue, WA and have a hybrid work environment.

Core AI Engineers will leverage a deep understanding of machine learning, software engineering, and problem solving to build scalable, robust solutions for clients. They will lead the entire development process, work with our econometricians, scientists, and engineering teams to scope solutions, develop and train ML models, deploy them into production environments, and set up monitors, alerts and diagnostic tools to support model operations.  

We are a tech focused consultancy where traditional reporting lines are fluid and adaptable, fostering an environment of collaborative teamwork. As a startup we seek candidates who are versatile, flexible, and eager to learn and share knowledge across a variety of roles and challenges. The most successful candidates will have a broad range of production software development skills, flexibility to wear many hats, and an enthusiasm for learning and a collaborative approach to sharing knowledge with colleagues at all levels. 

Role and Responsibilities: 

As a Core AI Staff Engineer, the successful candidate will be instrumental in developing and deploying cutting-edge machine learning models and software applications to deliver these models. The primary responsibilities will include: 

• Collaborating with cross-functional teams to identify business challenges and opportunities for applying machine learning techniques.
• Designing, implementing, and testing machine learning models to extract valuable insights from large and complex datasets
• Conducting thorough data analysis and pre-processing to ensure high-quality input for the models
• Integrating machine learning solutions into existing systems and processes and scaling them for real-world applications
• Contributing to the development of proprietary machine learning models/tools and frameworks
• Continuously improving CI/CD and testing frameworks and methodologies
• Continuously researching and staying updated on the latest advancements in machine learning and AI technologies

The ideal candidate:

• Possesses at least 4 years of experience in machine learning, software engineering, and system architecture, with demonstrated expertise in developing and deploying ML models and building infrastructure for model scaling and pipeline automation.
• Is proficient in programming languages such as Python, R, or Java, and has experience with machine learning libraries (e.g., TensorFlow, PyTorch, scikit-learn)
• Is familiar with cloud computing platforms and distributed computing frameworks
• Has experience implementing econometrics methods including double machine learning, time series analysis, and optimization techniques
• Is experienced with virtualization and cluster management tools, including Docker & Kubernetes
• Has demonstrated the ability to deliver end-to-end ML solutions that has meaningful value to stakeholders
• Is an excellent communicator and has strong teamwork skills to collaborate effectively with diverse stakeholders

Minimum Requirements:

• Bachelor’s degree in computer science, engineering, or a related scientific field
• 3+ years of experience as a Software Engineer with a focus on production systems at scale
• Proficiency in programming languages such as Python, R, or Java, C# and experience with machine learning libraries (e.g., TensorFlow, PyTorch, scikit-learn, Jax)
• 3+ years of experience developing and deploying machine learning models into production environments
• Familiarity with cloud computing platforms and distributed computing frameworks
• Strong problem-solving skills with the ability to work both independently and collaboratively in a dynamic environment

US Salary Range: $101,000 - $270,000, plus an annual discretionary bonus, 401k contribution, and competitive benefits package. Actual compensation within the range will depend upon the level the individual is hired into based on their skills, experience, qualifications.

At Keystone we believe diversity matters. At every level of our firm, we seek to advance and promote diversity, foster an inclusive culture, and ensure our colleagues have a deep sense of respect and belonging. If you are interested in growing your career with colleagues from varied backgrounds and cultures, consider Keystone Strategy.