Skip to content
mimi

Lead Data Scientist and Machine Learning Lead

Optum

Hyderabad · Hybrid Full-time Today

About the role

Primary Responsibilities

  • Lead end-to-end training and fine-tuning of Large Language Models (LLMs), including both open-source (e.g., Qwen, LLaMA, Mistral) and closed-source (e.g., OpenAI, Gemini, Anthropic) ecosystems
  • Architect and implement GraphRAG pipelines, including knowledge graph representation and retrieval for enhanced contextual grounding.
  • Design, train, and optimize semantic and dense vector embeddings for document understanding, search, and retrieval.
  • Develop semantic retrieval systems with advanced document segmentation and indexing strategies.
  • Build and scale distributed training environments using NCCL and InfiniBand for multi-GPU and multi-node training.
  • Apply reinforcement learning techniques (e.g., RLHF, RLAIF) to align model behavior with human preferences and domain-specific goals.
  • Collaborate with cross-functional teams to translate business needs into AI-driven solutions and deploy them in production environments
  • Comply with the terms and conditions of the employment contract, company policies and procedures, and any and all directives (such as, but not limited to,

Responsibilities

  • Lead end-to-end training and fine-tuning of Large Language Models (LLMs), including both open-source (e.g., Qwen, LLaMA, Mistral) and closed-source (e.g., OpenAI, Gemini, Anthropic) ecosystems
  • Architect and implement GraphRAG pipelines, including knowledge graph representation and retrieval for enhanced contextual grounding
  • Design, train, and optimize semantic and dense vector embeddings for document understanding, search, and retrieval
  • Develop semantic retrieval systems with advanced document segmentation and indexing strategies
  • Build and scale distributed training environments using NCCL and InfiniBand for multi-GPU and multi-node training
  • Apply reinforcement learning techniques (e.g., RLHF, RLAIF) to align model behavior with human preferences and domain-specific goals
  • Collaborate with cross-functional teams to translate business needs into AI-driven solutions and deploy them in production environments
  • Comply with the terms and conditions of the employment contract, company policies and procedures, and any and all directives (such as, but not limited to,

Skills

AnthropicGeminiGraphRAGInfiniBandLLaMALLMsMistralNCCLOpenAIRLAIFRLHFQwen

Don't send a generic resume

Paste this job description into Mimi and get a resume tailored to exactly what the hiring team is looking for.

Get started free