Senior​/Lead Machine Learning Engineer

About Salesforce

Salesforce is the #1 AI CRM, where humans with agents drive customer success together. Here, ambition meets action. Tech meets trust. And innovation isn't a buzzword - it's a way of life. The world of work as we know it is changing and we're looking for Trailblazers who are passionate about bettering business and the world through AI, driving innovation, and keeping Salesforce's core values at the heart of it all.

Ready to level-up your career at the company leading workforce transformation in the agentic era? You're in the right place! Agentforce is the future of AI, and you are the future of Salesforce.

We are seeking a highly skilled Machine Learning Engineer to design, build, and productionalize models that drive customer growth, engagement and retention. This role will focus specifically on attrition prediction and mitigation - identifying customers at risk of churn and surfacing proactive interventions that improve customer satisfaction and lifetime value.

You will work closely with data scientists, software engineers, product managers, and business stakeholders to build scalable ML systems that power attrition predictions, risk and mitigation explanations and next best action recommendations.

• Design predictive models for user and customer attrition using supervised, unsupervised, deep learning and generative techniques.
• Design scalable data pipelines for feature generation from both structured and unstructured sources of product adoption, sales activity, and customer engagement data (e.g. product telemetry, usage logs, CRM, sales activity, etc.).
• Build and maintain production-grade ML services, integrating models into APIs or decision systems that support real-time and batch use cases.
• Continuously monitor and improve model performance through drift detection, retraining automation and impact measurement.
• Collaborate with product and engineering teams to integrate models into production systems and agentic experiences, ensuring scalability, robustness and efficiency.
• Mentor junior engineers and data scientists and provide technical leadership in model architecture, experimentation, and deployment best practices.

• Demonstrated ability to take models from research to production
• Strong software engineering proficiency in Python and data manipulation skills like SQL.
• Experience using third-party and in-house Machine learning tools and infrastructure to develop reusable, high-performing Machine Learning systems, enable fast model development, low-latency serving and ease of model quality upkeep.
• Exposure to architectural patterns of a large, high-scale software application (e.g. well-designed APIs, high volume data pipelines, efficient algorithms, etc.)
• Familiarity with ML libraries such as scikit-learn, XGBoost, Pytorch, or Tensor Flow
• Experience with feature engineering on big data (Spark, Trino, Snowflake, etc.)
• Experience with ML lifecycle management tools (ML Flow, Airflow, Kubeflow or equivalents).
• Experience with containerization technologies (Docker) and orchestration (Kubernetes).
• Strong grasp of model evaluation, drift monitoring and explainability best practices.
• Experience with Agile development methodology, Test-Driven Development, incremental delivery, and CI/CD
• Experience owning and operating services throughout the software development lifecycle including design, development, release and maintenance.
• Experience communicating technical vision, mentoring junior engineers and managing projects.
• Experience developing and evaluating AI Agents that integrate with traditional ML models, (e.g. combining predictive scoring systems with generative or agentic workflows to automate customer engagement flows and recommendations).

• Familiarity with retention modeling or next best action recommendation systems.
• Experience developing or contributing to shared ML frameworks or internal ML Ops platforms.
• Experience with Feature Stores like Feast

When you join Salesforce, you'll be limitless in all areas of your life. Our benefits and resources support you to find balance and be your best, and our AI agents accelerate your impact so you can do your best. Together, we'll bring the power of Agentforce to organizations of all sizes and deliver amazing experiences that customers love. Apply today to not only shape the future - but to redefine what's possible - for yourself, for AI, and the world.

If you require assistance due to a disability applying for open positions please submit a request via this Accommodations Request Form.

Posting Statement

Salesforce is an equal opportunity employer and maintains a policy of non-discrimination with all employees and applicants for employment. What does that mean exactly? It means that at Salesforce, we believe in equality for all. And we believe we can lead the path to equality in part by creating a workplace that's inclusive, and free from…