Principal AI Architect - Supply Chain

We're building an AI-enabled supply chain that senses, predicts, prescribes, and acts. As Principal AI Architect, you'll design and build enterprise-grade AI systems — from data pipelines and models to agents and applications — that run in production across our Supply Chain organization. This is a hands-on technical role. You'll be in the architecture, in the code, and in the weeds of production systems. You'll design solutions, prototype approaches, write and review code, and unblock engineering teams building alongside you. Responsibilities Include But Not Limited To Architect and help build AI, generative AI, and agentic AI solutions — from proof of concept through production — using Python, FastAPI, PyTorch, LangGraph, and AutoGen Design solution architecture across data pipelines (Snowflake, Databricks), models, agents, RAG pipelines, vector databases, APIs, and React-based applications on AWS and Azure Get hands-on with complex technical problems: debugging production issues, prototyping new approaches, and reviewing code and system design Lead architecture reviews and technical decision-making, weighing tradeoffs across performance, cost, scalability, and maintainability Use GitHub, GitHub Actions, and GitHub Copilot to build and ship faster — for your own work and across teams Partner directly with product, data engineering, AI engineering, and platform teams to solve real technical problems, not just review their plans Mentor engineers through pairing, code review, and hands-on problem-solving Basic Qualifications Bachelor's degree in Computer Science, Engineering, Data Science, Information Systems, or related technical field 8+ years building production software, AI/ML systems, or data platforms 5+ years architecting and building production AI/ML solutions in Python, including LLMs, RAG architectures, and vector databases, on cloud-native infrastructure (AWS or Azure), with at least 1+ years of hands-on experience in agentic AI frameworks (e.g., LangGraph, AutoGen, or equivalent) Experience with modern MLOps practices and CI/CD (GitHub Actions or equivalent) Proven ability to take AI or software systems from concept through production, including debugging, performance tuning, and operational support Comfortable operating independently and making architecture calls with incomplete information Strong communication skills — able to explain technical tradeoffs to engineers, product partners, and senior leaders Preferred Qualifications Experience with FastAPI or similar frameworks for building production AI/ML services Experience with PyTorch for model development, fine-tuning, or inference Experience with Snowflake and/or Databricks for data pipelines feeding AI/ML systems Our Benefits — Designed With You In Mind Comprehensive Health & Well-being Coverage From your very first day, you’ll have access to medical, dental, vision, and prescription drug coverage — ensuring you and your family stay healthy and protected. Also, our Employee Assistance Program (EAP) offers confidential support for personal and professional challenges, always ready when you need it. Family Building Benefit We proudly support all paths to parenthood- including fertility and infertility treatments, adoption services, and gestational surrogacy. Generous Paid Time Off We believe in work-life balance. That’s why we offer: 17+ paid holidays, including shut-down from December 24th through New Years Day every year. Vacation, float & wellbeing days, sick time and fully paid parental leave when your family needs you most. Competitive Retirement Savings Plans We Help You Plan For The Future With An employer match on contributions to your 401k, Roth, and Catch-Up plans An employer contribution, even if you don’t contribute Income Protection & Insurance Options Benefit from both employer-provided and voluntary plan offerings, including life insurance, group accident, critical illness, etc. - supporting the needs of you and your family and ensuring peace of mind. Company Vehicle Lease Program Eligible employees and their immediate families can participate in the company vehicle lease program, providing access to Stellantis vehicles with insurance, maintenance, and unlimited miles included. Plus, take advantage of exclusive discounts on Stellantis products. Support for Your Growth and Giving Back We believe in investing in your future and your passions: Tuition reimbursement Student loan refinancing programs 18 paid volunteer hours each year to make a difference in your community And so much more! When you join us, you’re not just building a career — you’re joining a company that supports you, inside and outside of work. At Stellantis, we assess candidates based on qualifications, merit, and business needs. We welcome applications from all people without regard to sex, age, ethnicity, nationality, religion, sexual orientation, disability, or any characteristic protected by law. We believe that diverse teams reflect our identity as a global company, enabling us to better address the evolving needs of our customers and care for our future. EOE / Disability / Veteran Show more Show less