O
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