Join Keystone AI as a Machine Learning Engineer

Keystone is a distinguished consulting firm combining economics, technology, and strategy to solve intricate challenges faced by global brands. Our work impacts society on a global scale, operating at the forefront of influential technology cases that shape consumer behavior and regulatory frameworks. We provide value by delivering insights and products that define the future of the technological ecosystem. Keystone adopts an interdisciplinary approach, leveraging economics, software, technology, and business strategy to deliver transformative ideas.

At Keystone, we believe in seizing every opportunity for creativity. Transformative ideas often challenge common knowledge. We prioritize hard work, kindness, and trust, which can help navigate even catastrophic situations. Impact requires effort, and nothing is too far gone to be salvaged. We believe we can achieve unprecedented outcomes through unconstrained thinking.

Position Overview

As a Staff Machine Learning Engineer, you will apply your deep understanding of machine learning, software engineering, and problem-solving skills to build scalable, robust solutions for our clients. You will spearhead the entire development process, collaborating with our econometricians and engineering teams to scope solutions, train models, deploy them into production environments, and set up monitoring, alerts, and diagnostic tools to support ML model operations. We operate like a startup, valuing broad skill sets, flexibility, and enthusiasm for continuous learning and teaching.

Role and Responsibilities

Your primary responsibilities as a Machine Learning Engineer 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, complex datasets
• Conducting thorough data analysis and pre-processing to ensure high-quality inputs for the models
• Integrating machine learning solutions into existing systems 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
• Staying updated on the latest advancements in machine learning and AI technologies

Requirements

• Expert knowledge in system architecture and engineering best practices
• Proficiency in programming languages such as Python, R, or Java, and experience with machine learning libraries (e.g., TensorFlow, PyTorch, scikit-learn)
• Minimum 4 years of experience as a Machine Learning Engineer, Software Engineer, Data Scientist, or a similar role, focusing on developing and deploying ML models, and building infrastructure for model scaling and pipeline automation
• Familiarity with cloud computing platforms and distributed computing frameworks
• Experience with econometrics methods, including double machine learning, time series analysis, and optimization techniques
• Experience with virtualization and cluster management tools, including Docker and Kubernetes
• Proven track record of delivering end-to-end ML solutions that provide significant value to stakeholders
• Strong problem-solving skills and the ability to work independently in a dynamic environment
• Excellent communication and teamwork skills for effective collaboration with diverse stakeholders
• A strong academic background with a degree in Computer Science, Engineering, or a related field; additional academic credentials in Economics are preferred

Compensation and Benefits

The US salary range for this role is $175,000 - $235,000, plus an annual discretionary bonus, 401k contribution, and a competitive benefits package. Actual compensation within the range depends on the individual’s level, skills, experience, and qualifications.

At Keystone, we firmly believe that diversity matters. We strive to advance and promote diversity at every level of our firm, foster an inclusive culture, and ensure that all colleagues feel respected and valued. If you're interested in growing your career with colleagues from varied backgrounds and cultures, consider joining Keystone Strategy.