Optum is a global organization that delivers care, aided by technology to help millions of people live healthier lives. The work you do with our team will directly improve health outcomes by connecting people with the care, pharmacy benefits, data and resources they need to feel their best. Here, you will find a culture guided by inclusion, talented peers, comprehensive benefits and career development opportunities. Come make an impact on the communities we serve as you help us advance health optimization on a global scale. Join us to start Caring. Connecting. Growing together.
We are seeking a talented and passionate AI/ML Engineer to join our innovative technology team. The ideal candidate will be at the heart of designing and building the intelligent systems that power our products and services. This role involves working on the complete lifecycle of machine learning projects, from data collection and preprocessing to model development, deployment, and maintenance. You will be a key player in transforming complex data into actionable insights and creating AI-driven solutions that solve real-world problems.
Primary Responsibilities:
•Design and Develop ML Models: Research, design, and develop machine learning and deep learning models to address business challenges. This includes everything from selecting the right algorithms to building predictive models from the ground up
•Data Pipeline and Preprocessing: Build and maintain robust data pipelines. You will be responsible for gathering data from various sources, cleaning it, and transforming it into a usable format for model training
•Model Training and Evaluation: Train, test, and validate machine learning models to ensure they are accurate, reliable, and meet business requirements. This involves performing statistical analysis and running experiments to optimize model performance
•Deployment and MLOps: Deploy trained models into production environments and maintain their lifecycle. You will implement MLOps best practices for monitoring, retraining, and updating models to ensure they remain effective over time
•Collaboration and Innovation: Work closely with data scientists, software engineers, and product managers to integrate AI/ML models into our applications and systems
•Stay Current: Keep up-to-date with the latest advancements in the field of AI, machine learning, and deep learning, and be prepared to apply new research to our products
•Build Generative AI & LLM Workflows: Design, develop, and optimize LLM-powered applications and agentic workflows, including retrieval-augmented generation (RAG), prompt engineering, tool/function calling, and multi-step orchestration, to deliver production-grade generative AI features
•Ensure LLM Reliability & Evaluation: Establish evaluation frameworks, guardrails, and structured, type-safe output validation to ensure generative AI systems are accurate, safe, cost-efficient, and reliable in production
•Comply with all applicable Company policies, procedures, and business directives, changes including those relating to work location, team assignments, work schedules, and flexible work arrangements
Required Qualifications:
•Hands-on experience with ML frameworks and libraries such as TensorFlow, PyTorch, Keras, and Scikit-learn
•Hands-on experience integrating LLM provider APIs (e.g., Anthropic Claude, OpenAI, Azure OpenAI) into applications
•Experience building LLM workflows and agentic applications using orchestration frameworks such as LangGraph and LangChain
•Deep understanding of machine learning, deep learning, and statistical modeling concepts
•Familiarity with cloud platforms (e.g., AWS, Google Cloud, Azure) and their machine learning services (Databricks)
•Familiarity with core generative AI concepts including retrieval-augmented generation (RAG), prompt engineering, embeddings, and vector databases (e.g., Pinecone, FAISS, Chroma)
•Proficiency in data handling, including SQL, and experience with data processing tools like Pandas and NumPy
•Proficiency with Pydantic for data validation and enforcing structured, type-safe outputs from LLMs
•Proven excellent communication skills to articulate complex technical concepts to diverse audiences
•Demonstrated solid programming proficiency in languages like Python, R, or Java
•Demonstrated solid analytical and problem-solving abilities
•Demonstrated creative and proactive mindset with a drive for continuous improvement