Machine Learning Engineer

Role Overview

Design and implement machine learning models using Python and Spark to solve complex problems. Develop and maintain data pipelines using Airflow to ensure efficient data processing. Utilize MLflow for model tracking and management, ensuring seamless deployment in production environments. Collaborate with cross-functional teams to integrate machine learning solutions into existing systems. Optimize algorithms and workflows for performance and scalability on AWS and GCP platforms. Experiment with and implement state-of-the-art techniques in Transformers and LLMs to enhance product offerings.

Requirements

• 3–8 years of experience in machine learning engineering or related fields
• Proficiency in Python and experience with data processing frameworks like Spark
• Hands-on experience with MLOps tools such as MLflow and Weights & Biases
• Familiarity with cloud platforms, specifically AWS and GCP
• Strong understanding of CUDA for optimizing machine learning workflows

Benefits

• Competitive early-stage equity
• Salary range: $150K–$250K