Lead Machine Learning Engineer

This is a fully remote position, open to applicants in California.

📋 Description

• Work collaboratively with cross-functional teams, including Product Managers, Data Scientists, Data Engineers, Software Engineers, and Business units, to develop data and Machine Learning products.

• Assume responsibility for objectives and key results within your workstream, partnering with your manager to lead technical solutions.

• Design and construct robust systems for training, deploying, running inference, and monitoring Machine Learning and AI systems at scale.

• Advocate for code quality, reusability, scalability, maintainability, and security, while contributing to strategic architectural decisions.

• Establish processes and tools to guarantee data quality, enforce data governance policies, and uphold engineering best practices.

• Integrate Machine Learning and AI systems into production applications.

• Innovate with cutting-edge approaches, keeping up with the latest research and technologies within the broader ML engineering community.

⛳️ Requirements

• A completed BS, MS, or PhD in Computer Science, Mathematics, Statistics, Data Science, Engineering, Operations Research, or a related quantitative field.

• Over 7 years of experience as an engineer specializing in building Machine Learning systems.

• At least 2 years of technical leadership experience delivering machine learning solutions in collaboration with engineers, scientists, and business stakeholders.

• Proficient programming skills in Python and a solid understanding of fundamental computer science principles.

• Experience with machine learning and AI frameworks and libraries such as scikit-learn, HuggingFace, PyTorch, Tensorflow/Keras, MLlib, etc.

• Capable of designing, training, and evaluating machine learning and AI models while adhering to best practices like model selection, validation, bias/variance tuning, performance assessment, sensitivity analysis, and dimensionality reduction.

• Familiarity with MLOps practices, including automated model deployment, model performance monitoring, and data drift detection.

• Experience in building batch and streaming pipelines utilizing complex SQL, PySpark, Pandas, and similar frameworks.

• Knowledge of data warehouses (e.g., dimensional modeling), data lakes/Lakehouses, and other data architectures.

• Proficient in orchestrating complex workflows and data pipelines using Airflow or similar tools.

• Ability to load test deployed models at scale to pinpoint performance bottlenecks.

• Experience with Git, CI/CD pipelines, Docker, and Kubernetes.

• Skilled in architecting solutions on AWS or comparable public cloud platforms.

• Experience in developing data APIs, Microservices, and event-driven systems for integrating ML systems.

• Familiarity with Large Language Models (LLMs) and other generative AI methods, including their application in production.

• Experienced in assessing and implementing new data tools to improve the machine learning stack.

• Excellent interpersonal and verbal communication skills.

• Proven technical leadership experience with the capability to mentor and guide others.

🏝️ Benefits

• Paid time off (vacation, holidays, sick leave).

• Medical, dental, and vision insurance.

• 401(k) plan for eligible employees.

• Participation in long-term incentive programs.