1

Pytorch Developer Jobs in Ann Arbor, MI (NOW HIRING)

PyTorch), with a solid foundation in software engineering practices. * Experience with real-time systems or robotics, ideally with simulation- or vehicle-in-the-loop components. * Ability to lead ...

AI Engineer

Ann Arbor, MI

$150K - $250K/yr

And our Head of Engineering was one of the earliest engineers at Figma. AI Engineer ... Well-versed in using ML/NLP python packages such as tensorflow, pytorch, scikit-learn, transformers ...

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 ...

New

Senior Autonomy Engineer

Ann Arbor, MI

$102K - $140K/yr

Develop and maintain developer tooling and infrastructure, including build scripts, test harnesses ... Familiarity with ML/DL frameworks (e.g., PyTorch/TensorFlow) at a basic level. * Exposure to ...

Lead Research Engineer

Ann Arbor, MI · On-site +1

$100K - $132K/yr

... learn, PyTorch. * Take pride in writing clean, reusable, maintainable and well-tested code ... Proficiency in system analysis and design&Consider DevOps and automation as fundamental pillars of ...

... engineers, data scientists, and product teams to integrate models with business workflows. • ... TensorFlow, PyTorch). • Hands-on experience with PySpark for big data processing and model ...

Expertise in ML/DL development using PyTorch or TensorFlow, including experience with synthetic data generation, data curation, and model/algorithm evaluations. * Strong programming skills in Python ...

New

Senior Machine Learning Test Engineer

Novi, MI · On-site +1

$103K - $134K/yr

You are a quality-focused developer who is passionate about reliable, repeatable evaluation of ML ... Familiarity with ML frameworks (e.g., PyTorch, TensorFlow) * Experience with CI/CD tools and ...

Showing results 21-40

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 Ann Arbor, MI are hiring for Pytorch Developer jobs? Cities near Ann Arbor, MI with the most Pytorch Developer job openings:

Machine Learning Engineer, App SW

Wayve

Detroit, MI • Hybrid

$283K - $381K/yr

Full-time

Re-posted 12 days ago


Job description

The Role 

As an ML Engineer within the Application Engineering team, you'll lead critical initiatives that push the frontier of model-based autonomous driving-both in terms of core driving performance and feature-level intelligence such as personalisation, comfort, and collaboration.

You'll design and deliver ML-driven behaviors that scale from assisted to autonomous driving. Your work will span across model architecture, data pipelines, evaluation frameworks, and real-world deployment. You'll collaborate deeply with AI Platform, Simulation, Robot SW and Model Release teams to build systems that are performant, adaptable, and ready for production.

Responsibilities:
  • Develop and improve end-to-end driving models with state-of-the-art performance, robustness, and generalization.
  • Lead projects on personalized and collaborative driving, including behavior conditioning, comfort tuning, and user alignment.
  • Build evaluation pipelines and metrics for both closed-loop and open-loop driving performance and product readiness.
  • Curate and mine real-world and synthetic data to drive scenario diversity, coverage, and feature-specific development.
  • Influence architecture choices, training methodologies, and deployment pathways for production-scale learning systems.
  • Collaborate cross-functionally across various teams to ensure integration and iteration velocity.
  • Mentor senior engineers and shape the long-term technical direction across Autonomy.
About you: 

In order to set you up for success as a Machine Learning Engineer at Wayve, we're looking for the following skills and experience.  

Essential
  • Extensive and proven track record of shipping deep learning systems to production.
  • Expert in deep learning (esp. sequential models, control, planning, or perception).
  • Proficient in Python and other relevant languages (e.g. C++ and CUDA) and ML frameworks (esp. PyTorch), with a solid foundation in software engineering practices.
  • Experience with real-time systems or robotics, ideally with simulation- or vehicle-in-the-loop components.
  • Ability to lead technical initiatives across teams, drive alignment, and mentor engineers.
Desirable
  • Prior work in autonomous driving, imitation learning, or trajectory prediction.
  • Familiarity with personalization, human behavior modeling, or driver intent inference.
  • Experience integrating ML systems into production hardware or multi-agent simulation.

This role is a full-time role based in Sunnyvale or Detroit (hybrid) and the reasonably estimated salary for this role ranges from $283,500 to $381,600, plus a competitive equity package. Actual compensation is based on the candidate's skills, qualifications, and experience.

At Wayve we want the best of all worlds so we operate a hybrid working policy that combines time together in our offices and workshops to fuel innovation, culture, relationships and learning, and time spent working from home. We operate core working hours so you can determine the schedule that works best for you and your team.  

#LI-KM1