Demo

AI/ML Engineer

STAFFXPERT LLC
Milpitas, CA Contractor
POSTED ON 6/2/2026
AVAILABLE BEFORE 7/2/2026
Job Title: AI/ML Engineer

Location: Milpitas, CA (Hybrid 4 days onsite)

Duration: 6 Months

Job Summary

STAFFXPERT LLC is seeking an AI/ML Engineer on behalf of our client in Milpitas, CA to join a high-impact engineering team focused on building next-generation Generative AI systems.

In this role, you will transform advanced AI architectures into scalable, production-ready systems. You will work closely with senior AI architects to develop the core intelligence layer of an enterprise AI platform, with a strong emphasis on Retrieval-Augmented Generation (RAG) pipelines, multi-agent orchestration, and high-performance data systems.

Key Responsibilities RAG Pipeline Development

  • Design and implement end-to-end Retrieval-Augmented Generation (RAG) pipelines
  • Extract and process high-quality data from PDFs, wikis, FAQs, and enterprise knowledge sources
  • Develop advanced chunking strategies and metadata enrichment techniques
  • Improve retrieval accuracy and relevance through iterative tuning

Vector Database & Search Systems

  • Work with vector databases such as Qdrant, Pinecone, or Weaviate
  • Design hybrid search systems combining keyword and semantic retrieval
  • Implement efficient indexing strategies (e.g., HNSW) and optimize similarity search performance

Multi-Agent Systems

  • Build agentic workflows using frameworks such as LangGraph, Agno, or Google ADK
  • Develop tools, state management systems, and inter-agent communication logic
  • Orchestrate multi-agent pipelines for complex business automation use cases

Data Engineering for AI

  • Build scalable ETL pipelines for unstructured and semi-structured data
  • Transform raw content into structured, searchable knowledge bases
  • Maintain data quality, versioning, and governance standards

Evaluation & Optimization

  • Evaluate system performance using frameworks such as RAGAS or LangSmith
  • Optimize prompts, retrieval pipelines, and orchestration logic
  • Reduce hallucinations and improve response relevance and consistency

Deployment & Scaling

  • Containerize applications using Docker
  • Deploy low-latency, production-grade AI services
  • Ensure scalability, reliability, and performance optimization in production environments

Technical Requirements

  • Strong proficiency in Python (including Asyncio, FastAPI, and Pydantic)
  • Hands-on experience with agentic AI frameworks (LangGraph, Agno, or Google ADK)
  • Experience with vector databases such as Qdrant, Weaviate, Pinecone, or similar
  • Strong understanding of similarity search, embeddings, and HNSW indexing
  • Experience building RAG systems and working with LLM-based applications
  • Strong background in processing and structuring unstructured data sources
  • Familiarity with production deployment practices for AI systems

Hourly Wage Estimation for AI/ML Engineer in Milpitas, CA
$67.00 to $86.00
If your compensation planning software is too rigid to deploy winning incentive strategies, it’s time to find an adaptable solution. Compensation Planning
Enhance your organization's compensation strategy with salary data sets that HR and team managers can use to pay your staff right. Surveys & Data Sets

What is the career path for a AI/ML Engineer?

Sign up to receive alerts about other jobs on the AI/ML Engineer career path by checking the boxes next to the positions that interest you.
Income Estimation: 
$101,387 - $124,118
Income Estimation: 
$119,030 - $151,900
Income Estimation: 
$101,387 - $124,118
Income Estimation: 
$119,030 - $151,900
Employees: Get a Salary Increase
View Core, Job Family, and Industry Job Skills and Competency Data for more than 15,000 Job Titles Skills Library

Job openings at STAFFXPERT LLC

  • STAFFXPERT LLC York, NY
  • Title: Technical Project Manager (Capital Markets) Location: New York, NY (Hybrid – 4 days onsite) Duration: 6 Months Job Summary STAFFXPERT LLC is seeking... more
  • 2 Days Ago

  • STAFFXPERT LLC Sunnyvale, CA
  • Job Title: Atlassian Expert (Operations & Projects) Location: Sunnyvale, CA or Bellevue, WA (Hybrid 3 Days Onsite) Employment Type: Contract (6 Months) Job... more
  • 4 Days Ago

  • STAFFXPERT LLC Winchester, VA
  • Business Analyst JD Edwards (JDE) Location: Winchester, VA Employment Type: Contract Job Summary STAFFXPERT LLC is seeking a Business Analyst JD Edwards (J... more
  • 11 Days Ago

  • STAFFXPERT LLC Chicago, IL
  • AWS Data Engineer Location Hybrid – Chicago, IL / Lafayette, LA / Nashville, TN / Richardson, TX Job Summary STAFFXPERT LLC is seeking an AWS Data Engineer... more
  • 12 Days Ago


Not the job you're looking for? Here are some other AI/ML Engineer jobs in the Milpitas, CA area that may be a better fit.

  • AI Fund Mountain View, CA
  • Mountain View, CA LandingAI – Engineering / Full time / On-site LandingAI is building the infrastructure to make the world’s documents computable. We are a... more
  • 3 Days Ago

  • Vibotek LLC Mountain View, CA
  • Job Title: Android AI/ML Engineer - Infrastructure (AI ML Ops Engineer III) Job Category: Technical Position Summary We seek an experienced Android AI/ML E... more
  • 28 Days Ago

AI Assistant is available now!

Feel free to start your new journey!