Pytorch Developer Jobs in Minnesota (NOW HIRING)

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 highly skilled and innovative AI/ML Engineer with deep expertise in Generative AI technologies to develop and deploy cutting-edge solutions within the Pharmacy domain. The ideal candidate will have hands-on experience with Large Language Models (LLMs), Retrieval-Augmented Generation (RAG), Agentic AI, and frameworks such as LangGraph and LangChain. Solid proficiency in Python and experience with deep learning frameworks like TensorFlow, PyTorch, or Keras are essential.

This role will involve close collaboration with Pharmacy Operations and Capability teams. On the Operations side, you'll engage directly with Pharmacists to understand and solve real-world challenges. On the Capability side, you'll work with Capability leaders to identify problem areas and co-design AI/ML solutions. You'll also contribute to broader initiatives in Pharmacy Modernization, aligning with organizational goals to leverage emerging AI technologies for reducing manual work, accelerating delivery, and improving Net Promoter Score (NPS).

You'll enjoy the flexibility to work remotely * from anywhere within the U.S. as you take on some tough challenges. For all hires in the Minneapolis or Washington, D.C. area, you will be required to work in the office a minimum of four days per week.

Primary Responsibilities:

• Design, develop, and deploy AI/ML models using GenAI technologies (LLMs, RAG, Agentic AI)
• Utilize frameworks such as LangGraph, LangChain, TensorFlow, PyTorch, or Keras for model development
• Deploy AI/ML solutions in production environments, preferably on cloud platforms like Azure
• Conduct rigorous model testing, validation, and performance optimization
• Containerize AI/ML projects and implement CI/CD pipelines using tools like GitHub Actions
• Collaborate with stakeholders to preprocess, analyze, and interpret large-scale datasets
• Stay current with emerging trends and advancements in generative AI and ML technologies
• Ensure code quality and security by identifying and resolving vulnerabilities
• Maintain solid programming practices with a focus on Python
• Work Eastern Time Zone hours to support collaboration across a global team
• Participate in on-call rotations as scheduled
• Not a research focused role, but more of an AI/ML product development role in a software engineering team

You'll be rewarded and recognized for your performance in an environment that will challenge you and give you clear direction on what it takes to succeed in your role as well as provide development for other roles you may be interested in.

Required Qualifications:

• BS in Computer Science, Engineering, or related technical field (or equivalent experience)
• 5 years of industry experience in AI/ML or Data Science
• 3 years of hands-on experience with Python and version control tools like GitHub
• 3 years of experience deploying AI/ML solutions in production environments
• Experience with Git management and CI/CD best practices
• Experience in designing high-level cloud architecture and solution designs for AI/ML solutions
• Should have experience of AI/ML projects deployed in Production
• Required Qualification: Must be authorized to work in the United States without the need for current or future employer-sponsored visa sponsorship (e.g., H-1B, TN, F-1/OPT, CPT, or other employment-based visa status).

Preferred Qualifications:

• Certifications in AI/ML or related technologies
• Experience with deep learning frameworks (e.g., PyTorch, TensorFlow)
• Knowledge of UAIS and MLRB
• Familiarity with cloud platforms (Azure, AWS, Google Cloud)

*All employees working remotely will be required to adhere to UnitedHealth Group's Telecommuter Policy

Pay is based on several factors including but not limited to local labor markets, education, work experience, certifications, etc. In addition to your salary, we offer benefits such as, a comprehensive benefits package, incentive and recognition programs, equity stock purchase and 401k contribution (all benefits are subject to eligibility requirements). No matter where or when you begin a career with us, you'll find a far-reaching choice of benefits and incentives. The salary for this role will range from $120,100 - $214,500 annually based on full-time employment. We comply with all minimum wage laws as applicable.

Application Deadline: This will be posted for a minimum of 2 business days or until a sufficient candidate pool has been collected. Job posting may come down early due to volume of applicants.

At UnitedHealth Group, our mission is to help people live healthier lives and make the health system work better for everyone. We believe everyone-of every race, gender, sexuality, age, location and income-deserves the opportunity to live their healthiest life. Today, however, there are still far too many barriers to good health which are disproportionately experienced by people of color, historically marginalized groups and those with lower incomes. We are committed to mitigating our impact on the environment and enabling and delivering equitable care that addresses health disparities and improves health outcomes - an enterprise priority reflected in our mission.    

UnitedHealth Group is an Equal Employment Opportunity employer under applicable law and qualified applicants will receive consideration for employment without regard to race, national origin, religion, age, color, sex, sexual orientation, gender identity, disability, or protected veteran status, or any other characteristic protected by local, state, or federal laws, rules, or regulations.

UnitedHealth Group is a drug - free workplace. Candidates are required to pass a drug test before beginning employment.