Machine Learning Engineer, RAG

To get the best candidate experience, please consider applying for a maximum of 3 roles within 12 months to ensure you are not duplicating efforts.

Job Category

Software Engineering

Job Details

About Salesforce

We’re Salesforce, the Customer Company, inspiring the future of business with AI+ Data +CRM. Leading with our core values, we help companies across every industry blaze new trails and connect with customers in a whole new way. And, we empower you to be a Trailblazer, too — driving your performance and career growth, charting new paths, and improving the state of the world. If you believe in business as the greatest platform for change and in companies doing well and doing good – you’ve come to the right place.

Machine Learning Engineer - RAG
This role involves building the next-gen Retrieval-Augmented Generation (RAG) by demonstrating sophisticated generative AI services, pipelines, and components to drive the development and delivery of Agentforce. You will work on and ship impactful generative AI platforms, applications, and products used by millions of people every day.
The Role
You will play a critical role in integrating artificial intelligence and generative AI into sophisticated RAG technologies, enterprise Knowledge Graphs, and the latest LLM algorithms and technologies to build the next wave of intelligent agents. You will participate in the end-to-end AI product development lifecycle, designing and developing scalable deep learning and generative AI systems and services. You will collaborate with AI Platform Engineers and Software Engineers to define requirements and develop reusable workflows and ML pipelines.
What You’ll Do:
- Design and deliver scalable RAG services that can be integrated with numerous applications, support thousands of tenants, and operate at scale in production.
- Drive system efficiencies through automation, including capacity planning, configuration management, performance tuning, monitoring, and root cause analysis.
- Participate in periodic on-call rotations and be available to resolve critical issues.
- Collaborate with Product Managers, Application Architects, Data Scientists, and Deep Learning Researchers to understand customer requirements, design prototypes, and bring innovative technologies to production.
- Engage in fun-spirited yet thought-provoking conversations with your team around fun topics, such as: Should the popularity of penguins influence conservation efforts? Is creativity more like a neural network or a heuristic algorithm? How would society evolve if humans communicated through colors as well as words?
- Brainstorm with Product Managers, Designers, and Engineers to conceptualize and build new features for our large (and growing!) user base.
- Produce high-quality results by leading or collaborating heavily to large multi-functional projects that have a significant impact on the business.
- Help other engineers actively own features or systems and define their long-term health, while also improving the health of surrounding systems.
- Assist our skilled support team and operations team in triaging and resolving production issues.
- Mentor other engineers and deeply review code.
- Improve engineering standards, tooling, and processes.
Required Skills:
- 10+ years of proven experience in ML engineering, building AI systems and/or services.
- Strong proficiency in NLP and machine learning models.
- Experience with LLMs and prompt engineering.
- Proven experience building and applying machine learning models to business applications.
- Proven track record to innovate and deliver results at scale.
- Experience with distributed, scalable systems and modern data storage, messaging, and processing frameworks, such as Kafka, Spark, Docker, and Hadoop.
- Proven understanding of deep learning and machine learning algorithms.
- Grit, drive, and a strong sense of ownership, coupled with teamwork and leadership skills.
- Experience with functional or imperative programming languages: PHP, Python, Ruby, Go, C, Scala, or Java.
- Built with common ML frameworks like PyTorch, TensorFlow, Keras, XGBoost, or Scikit-learn.
- Experience building batch data processing pipelines with tools like Apache Spark, SQL, Hadoop, EMR, MapReduce, Airflow, Dagster, or Luigi.
- An analytical and data-driven approach, and know how to measure success with complicated ML/AI products.
- Led technical architecture discussions and helped drive technical decisions within the team.
- The ability to write understandable, testable code with an eye towards maintainability.
- Strong communication skills and you are capable of explaining sophisticated technical concepts to designers, support, and other specialists.
- Strong computer science fundamentals: data structures, algorithms, programming languages, distributed systems, and information retrieval.
- A bachelor’s degree in Computer Science, Engineering, Statistics, Mathematics, or a related field, or you have equivalent training, fellowship, or work experience.
Preferred Skills:
- Expertise in retrieval systems and search algorithms.
- Familiarity with vector databases and embeddings.
- Strong background in a wide range of ML approaches, from Artificial Neural Networks to Bayesian methods.
- Experience with conversational AI.
- Excellent problem-solving skills; the ability to take on problems the world has yet to solve.
- Demonstrated track record of cultivating strong working relationships and driving collaboration across multiple technical and business teams.
- Knowledge of using multiple data types in RAG solutions including structured, unstructured, and graph.

Accommodations

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 discrimination. Know your rights: workplace discrimination is illegal. Any employee or potential employee will be assessed on the basis of merit, competence and qualifications – without regard to race, religion, color, national origin, sex, sexual orientation, gender expression or identity, transgender status, age, disability, veteran or marital status, political viewpoint, or other classifications protected by law. This policy applies to current and prospective employees, no matter where they are in their Salesforce employment journey. It also applies to recruiting, hiring, job assignment, compensation, promotion, benefits, training, assessment of job performance, discipline, termination, and everything in between. Recruiting, hiring, and promotion decisions at Salesforce are fair and based on merit. The same goes for compensation, benefits, promotions, transfers, reduction in workforce, recall, training, and education.

In the United States, compensation offered will be determined by factors such as location, job level, job-related knowledge, skills, and experience. Certain roles may be eligible for incentive compensation, equity, and benefits. Salesforce offers a variety of benefits to help you live well including: time off programs, medical, dental, vision, mental health support, paid parental leave, life and disability insurance, 401(k), and an employee stock purchasing program. More details about company benefits can be found at the following link: https://www.salesforcebenefits.com.Pursuant to the San Francisco Fair Chance Ordinance and the Los Angeles Fair Chance Initiative for Hiring, Salesforce will consider for employment qualified applicants with arrest and conviction records.

For California-based roles, the base salary hiring range for this position is $172,000 to $384,100.