Milestone Technologies, Inc.

Senior AI/ML Engineer

Milestone Technologies, Inc.  •  Onsite  •  2 hours ago
Apply
AI can make mistakes so check important info. Chat history is never stored.

Job Description

Role summary

Implements and tests assigned capabilities across feature suites, including RAG pipeline integration and prompt tuning; delivers GraphRAG-based dynamic query generation, graph-aware retrieval, and knowledge-graph schema evolution support; delivers full orchestration integration for the platform's orchestration component.

Key responsibilities

  • Develop feature logic for assigned capabilities within assigned feature suites
  • Design and tune prompts and RAG pipeline integrations
  • Validate sprint-level acceptance criteria against the golden dataset baseline
  • Coordinate with the AI Architect and Backend Engineer on design and integration questions
  • Build GraphRAG-based dynamic query generation to replace template-based SPARQL querying
  • Deliver graph-aware retrieval agent cards
  • Support knowledge-graph schema evolution
  • Coordinate with the KG validation workstream on schema and validation questions
  • Deliver full orchestration integration with the platform's cookbook-based orchestration component
  • Integrate the RAG-layer retrieval pipeline to inform agent selection, routing and response generation
  • Document the upgrade path to fine-tuning (documentation only, not an implementation, in this phase)
  • Expand the golden dataset and semantic grounding
  • Maintain question-and-answer baseline categorization

Required skills & experience

  • 6+ years AI engineering experience with strong Python fundamentals
  • Hands-on RAG pipeline and prompt-engineering experience
  • LLM application development experience
  • Comfortable validating work against a capped, governed golden dataset rather than open-ended tuning
  • 3+ years RAG / GraphRAG implementation experience
  • Vector store integration experience
  • Familiarity with SPARQL and graph-aware retrieval
  • Strong Python skills
  • 3+ years RAG implementation and orchestration integration experience
  • Vector store / retrieval pipeline integration experience
  • Comfortable scoping a fine-tuning upgrade path as a design recommendation rather than an implementation
  • 3+ years NLP / semantic engineering experience
  • Ontology mapping and semantic disambiguation experience
  • Experience building or validating golden Q&A datasets for grounding LLM systems

Preferred qualifications

  • Experience on a regulated or enterprise AI product
  • Familiarity with LangGraph or comparable agent frameworks
  • Direct experience replacing a template-based SPARQL approach with GraphRAG
  • Experience coordinating with a separate KG validation team
  • Experience with cookbook-style orchestration frameworks
  • Experience writing forward-looking technical design documents (vs. shipping code) when scope calls for it
  • Experience working within a per-sprint expansion cap rather than open-ended dataset growth
  • Pharma/manufacturing terminology disambiguation experience

Education

B.Tech/MCA in Computer Science, AI/ML, or a related field, or equivalent practical experience.

Milestone Technologies, Inc.

About Milestone Technologies, Inc.

Milestone Technologies is a global IT Services and Digital Solutions company based in Silicon Valley that helps hundreds of leading corporations deliver technology around the globe.

We work with the world’s leading companies to deliver services and technologies at scale, accelerate digital operations, develop innovative applications, and drive efficiencies throughout their organization.

Milestone is focused on building an employee-first, performance-based culture, and for over 25 years, we have demonstrated a history of supporting category-defining enterprise clients that are growing ahead of the market. The company specializes in providing solutions across Application Services and Consulting, Digital Product Engineering, Digital Workplace Services, Private Cloud Services, AI/Automation, and ServiceNow.

Milestone culture is built to provide a collaborative, inclusive environment that supports employees and empowers them to reach their full potential.

Follow us on:

Facebook: https://www.facebook.com/MilestoneTechnologiesInc

Twitter: https://twitter.com/MilestoneTech

Blog: https://milestone.tech/blog/

Industry
IT & Software
Company Size
1,001-5,000 employees
Headquarters
Fremont, California
Year Founded
1997
Social Media