Staff ML Engineer

Our company is an AI-driven Growth and Personalization Platform that is transforming how businesses engage, retain, and monetize their customers. Our software provides real-time personalization using machine learning to treat each customer as an individual, creating a “cohort-of-one.” With our platform, numerous personalized campaigns can be quickly generated and deployed, eliminating outdated, rules-based segmentation and targeting. Our solution is user-friendly, integrates with major data and customer engagement tools, and optimizes internal resource allocation.

Our team was founded on our shared expertise in driving product growth at leading tech companies. We recognized a shift where successful teams achieved significant revenue growth and cost efficiency by moving from static segmentation to real-time, machine learning-driven decision-making.
About the Role
As a key member of our Data Science team, you will significantly influence the data science vision and roadmap, focusing on building and enhancing our Machine Learning models and AI products. Your research will span recommender systems, causal inference, transformers, foundational models, content understanding, and reinforcement learning.

You will develop production-grade models that serve decisions to millions of users daily, working at the crossroads of artificial intelligence, process automation, and workflow optimization to create intelligent agents capable of understanding objectives, making decisions, and adapting to real-time changes. Additionally, you will analyze customer data to derive insights and collaborate across engineering, product, and business teams.
Responsibilities

• Design, implement, and research machine learning algorithms to address growth and personalization challenges while continuously refining them.
• Develop AI-driven autonomous agents to manage complex workflows across various applications and objectives.
• Enhance offline model evaluation methods.
• Analyze large datasets to extract knowledge and communicate insights relevant to critical business questions using traditional statistical or machine learning methods.
• Present data-driven insights and recommendations to product, growth, and data science teams within client organizations.
• Stay informed on ML research through literature reviews, conferences, and networking.

Qualifications

• Master’s or PhD in Computer Science, Artificial Intelligence, Machine Learning, or a related field.
• 5+ years of experience in training, improving, and deploying ML models, and developing production software systems like data pipelines or dashboards, preferably in rapidly growing tech companies.
• Proficiency in ML frameworks (TensorFlow/PyTorch/Jax) and the Python data science ecosystem (NumPy, SciPy, Pandas, etc.).
• Familiarity with LLMs, RAG, and information retrieval in building production-grade solutions.
• Experience with recommender systems, feed ranking, and optimization.
• Knowledge of causal inference techniques and experimental design.
• Experience working with cloud services and managing large-scale datasets.
• Ability to thrive in ambiguity, demonstrate an ownership mindset, and willingness to engage in responsibilities beyond your core role (e.g., product vision, strategic roadmap, engineering, etc.).