Skip to main content

Machine Learning Engineer, RAG

a couple of women smiling
  • Software Engineering
    • California - San Francisco
    • California - Palo Alto
  • Full time
  • JR266431
  • California Salary $172,000 - $384,100

Description

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.

For roles in San Francisco and Los Angeles: 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.

At Salesforce, we strive to create an accessible and inclusive experience for all candidates.

If you need a reasonable accommodation during the application or the recruiting process, please submit a request via this Accommodation Request Form.

Please note that Salesforce uses an automated employment decision tool to help our recruiters assess and evaluate candidates’ resumes. If you do not want Salesforce to use this tool with your application, please submit a request via this form.

Equal Opportunity Statement.

At Salesforce we believe that the business of business is to improve the state of our world. Each of us has a responsibility to drive Equality in our communities and workplaces. We are committed to creating a workforce that reflects society through inclusive programs and initiatives such as equal pay, employee resource groups, inclusive benefits, and more. Learn more about Equality at www.equality.com and explore our company benefits at www.salesforcebenefits.com.

Salesforce is an equal employment opportunity and affirmative action employer. Qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender perception or identity, national origin, age, marital status, protected veteran status, or disability status. Know your rights: workplace discrimination is illegal. Salesforce does not accept unsolicited headhunter and agency resumes. Salesforce will not pay any third-party agency or company that does not have a signed agreement with Salesforce.

Salesforce welcomes all.

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. More details about our company benefits can be found at the following link: https://www.salesforcebenefits.com.

We can't wait to meet you!

Join our Talent Community and be the first to know about open roles, career tips, events happening near you, and much more.

Join our Talent Community
Ae Hiring Cta Banner