Senior Staff Software Engineer, Machine Learning (Multiple Levels/ Principal, Architect)

$165,600 - $372,900/Yr

Salesforce - Atlanta, GA

Full-time - Senior

Atlanta, GA

Publishing Industries

About the position

The Senior Staff Machine Learning Engineer at Salesforce will be responsible for crafting and implementing machine learning and generative AI powered features that enhance the user experience. This role involves collaborating with cross-functional teams to conceptualize new features, leading significant projects, mentoring engineers, and driving system efficiencies through automation. The ideal candidate will have extensive experience in machine learning and software engineering, particularly in deploying models at scale.

Responsibilities

• Brainstorm with Product Managers, Designers, and Engineers to conceptualize and build new features.
• Produce high-quality results by leading or contributing heavily to large multi-functional projects.
• Help other engineers actively own features or systems and define their long-term health.
• Collaborate with peers across Engineering to triage bugs and troubleshoot production issues.
• Mentor other engineers and deeply review code.
• Improve engineering standards, tooling, and processes.
• Design and deliver scalable RAG services that can be integrated with numerous applications.
• Drive system efficiencies through automation, including capacity planning and performance tuning.
• Participate in periodic on-call rotations to resolve critical issues.
• Collaborate with Product Managers, Application Architects, Data Scientists, and Deep Learning Researchers.

Requirements

• 8+ years experience with machine learning and software engineering.
• Experience deploying machine learning models and generative AI at scale, especially for text-based applications.
• Proficiency in functional or imperative programming languages such as PHP, Python, Ruby, Go, C, Scala, or Java.
• Experience with common ML frameworks like PyTorch, Keras, XGBoost, TensorFlow, or Scikit-learn.
• Strong analytical and data-driven approach to measuring success with ML/AI products.
• Experience leading technical architecture discussions and driving technical decisions.
• Ability to write understandable, testable, and maintainable code.
• Strong communication skills to explain technical concepts to non-technical stakeholders.
• Strong computer science fundamentals including data structures, algorithms, and distributed systems.

Nice-to-haves

• Experience deploying production RAG pipelines.
• Experience in A/B testing and experimentation.
• Experience with LLM evaluation and monitoring at scale.
• Experience with search or ranking-oriented ML features.
• Strong background in various ML approaches, including Artificial Neural Networks and Bayesian methods.
• Experience with conversational AI and retrieval systems.
• Familiarity with vector databases and embeddings.
• Knowledge of handling multiple data types in RAG solutions.
• Experience building batch data processing pipelines with tools like Apache Spark, SQL, or Airflow.
• Familiarity with search technologies like Elasticsearch and Solr.

Benefits

• Health insurance
• Dental insurance
• Vision insurance
• 401k
• 401k matching
• Paid holidays
• Flexible scheduling
• Professional development
• Tuition reimbursement
• Employee discount programs