1

Pytorch Developer Jobs in Birmingham, MI (NOW HIRING)

Technical Environment Programming: Python, R AI/ML Frameworks: TensorFlow, PyTorch, Keras, Scikit-learn Big Data: Apache Spark, Databricks Cloud Platforms: AWS (SageMaker, EC2, S3), Microsoft Azure ...

Technical Environment Programming: Python, R AI/ML Frameworks: TensorFlow, PyTorch, Keras, Scikit-learn Big Data: Apache Spark, Databricks Cloud Platforms: AWS (SageMaker, EC2, S3), Microsoft Azure ...

Senior Engineer, AI

Novi, MI · On-site

$90.75 - $133.10/hr

Engineer audio systems and integrated technology platforms that augment the driving experience ... Familiarity with frameworks and tooling such as LangChain, LlamaIndex, Transformers, PyTorch ...

AI Specialist

Pontiac, MI · On-site

$80/hr

Programming: Python, R * ML Frameworks: TensorFlow, PyTorch, Keras, Scikit-learn * Big Data & Analytics: Apache Spark, Databricks, Jupyter Notebooks * Cloud Platforms: AWS (SageMaker, EC2, S3), Azure ...

We are seeking a Principal AI Engineer with deep, hands‑on experience in Large Language Models ... Strong proficiency in Python and modern AI/ML frameworks and libraries (e.g., PyTorch, TensorFlow ...

Experience with ML frameworks (Scikit-learn, TensorFlow, PyTorch) * Hands-on experience with LLMs, prompt engineering, and Generative AI * Experience working with cloud platforms (GCP and/or AWS)

AI Data Engineer

Detroit, MI

$113K - $136K/yr

Hands-on experience with ML frameworks like TensorFlow, PyTorch, and Scikit-learn, as well as LLM ... * DevOps and MLOps: Experience with CI/CD, Docker, and ML lifecycle management tools like MLflow ...

Data / BI Architect

Pontiac, MI · On-site

$63.25 - $81.50/hr

... Programming for data visualizations, Python, TensorFlow, PyTorch, Keras, Scikit-learn, Apache Spark, Databricks, Jupyter Notebooks, AWS (SageMaker, EC2, S3), Azure (Machine Learning Studio ...

Showing results 41-60

Pytorch Developer information

What is a PyTorch developer?

A PyTorch Developer is a software engineer or data scientist who specializes in using PyTorch, an open-source machine learning library, to build and deploy deep learning models. Their responsibilities typically include designing neural network architectures, training and evaluating models, and optimizing code for performance. PyTorch Developers work in fields such as artificial intelligence, computer vision, and natural language processing, collaborating with teams to solve complex problems using machine learning. They are proficient in Python and have a strong understanding of deep learning concepts. Additionally, they often contribute to research, development, and the deployment of AI solutions in production environments.

What are the key skills and qualifications needed to thrive as a PyTorch developer, and why are they important?

To thrive as a Pytorch Developer, you need strong programming skills in Python, a solid grasp of machine learning concepts, and experience with deep learning frameworks—especially PyTorch itself. Familiarity with tools like CUDA, Jupyter Notebooks, and version control systems (e.g., Git) is typically expected, along with knowledge of cloud platforms or relevant certifications. Problem-solving ability, effective collaboration, and clear communication are crucial soft skills for success in this role. These skills and qualities are vital for efficiently building, optimizing, and deploying machine learning models in real-world applications.

What is the difference between Pytorch Developer vs Machine Learning Engineer?

AspectPytorch DeveloperMachine Learning Engineer
Required CredentialsBachelor's or higher in CS, experience with PyTorchBachelor's or higher in CS, data science, or related field, with ML experience
Work EnvironmentResearch labs, AI startups, tech companies focusing on deep learningTech companies, finance, healthcare, often involving deployment and scaling ML models
Industry UsagePrimarily in AI research and development teamsAcross industries implementing ML solutions in production

While both roles require knowledge of machine learning and experience with PyTorch, a Pytorch Developer mainly focuses on developing and optimizing deep learning models using PyTorch. A Machine Learning Engineer often has a broader scope, including deploying, maintaining, and scaling ML models across various platforms and industries.

What are some common challenges PyTorch developers face when deploying machine learning models to production environments?

Pytorch Developers often encounter challenges when transitioning models from research to production, such as optimizing model performance for inference speed and memory usage, ensuring compatibility with deployment frameworks like TorchScript or ONNX, and managing dependencies across different systems. Additionally, integrating PyTorch models into existing software stacks and maintaining reproducibility can be complex. Collaborating closely with DevOps and data engineering teams is crucial to address these issues and ensure smooth deployment.

What cities near Birmingham, MI are hiring for Pytorch Developer jobs?

Cities near Birmingham, MI with the most Pytorch Developer job openings:

Infographic showing various Pytorch Developer job openings in Birmingham, MI as of August 2026, with employment types broken down into 79% Full Time, 3% Part Time, and 18% Contract. Highlights an 81% Physical, 4% Hybrid, and 15% Remote job distribution.

Principal AI Architect - Supply Chain

Stellantis

Auburn Hills, MI • On-site

Full-time

Medical, Dental, Vision, Retirement, PTO

Posted 4 days ago


Stellantis rating

7.5

Company rating: 7.5 out of 10

Based on 130 frontline employees who took The Breakroom Quiz

14th of 44 rated automakers


Job description

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.

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 included and optional disability, life, and other insurance programs — because your peace of mind matters.

Company Vehicle Lease Program
Eligible employees and their immediate families can enjoy company vehicle lease options with included insurance, maintenance, and unlimited mileage. Plus, take advantage of exclusive discounts on Stellantis products.

Family Building Benefit
We proudly support all paths to parenthood- including fertility and infertility treatments, adoption services, and gestational surrogacy.

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.


What Stellantis employees say

Pay

Benefits

Hours and flexibility

Workplace

Get the full story on Breakroom