Hiring: Lead AI/ML Engineer Company: Optum India Location: Hyderabad, Telangana, India Job Posted Time: 2026-09-16 16:41:27 Employment Type: Contract / Hybrid Target Skills & Keywords : AI, AWS, Azure, Deep Learning, EDA, Fine-tuning, LLM, Large Language Model, Machine Learning, Milvus, NLP, Node.js, PyTorch, Python, Reinforcement Learning, SQL, Weaviate About the job Experience: •15+ years of experience in applied AI/ML with statistics, with a solid track record of delivering production-grade models Required Skills: •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. •Partner cross-functionally with cross-functional teams to translate business needs into AI-driven solutions and deploy them in production environments Qualifications: •PhD or master's degree in computer science, Machine Learning, or related field •Deep expertise in: NLP, Fundamental machine learning, deep learning, transformer, state space-based architecture •Solid knowledge of Python coding, SQL and database queries, data preparation, and analysis •Knowledge of Azure ML and/or AWS •Knowledge of exploratory Data Analysis (EDA) •Knowledge of LLM training and fine-tuning (e.g., GPT, LLaMA, Mistral, Qwen) •Knowledge of graph-based retrieval systems (GraphRAG, knowledge graphs) •Knowledge of embedding models (e.g., BGE, E5, SimCSE) •Knowledge of semantic search and vector databases (e.g., FAISS, Weaviate, Milvus) •Knowledge of document segmentation and preprocessing (OCR, layout parsing) Compensation: •Flexible work environment (work from home / hybrid options) •Competitive benefits and rewards package Interested candidates, please apply directly through the job posting on company's career page or try via AI auto apply on this platform. Don't miss this opportunity to join a forward-thinking team!