1

Pytorch Developer Jobs in Madison, WI (NOW HIRING)

Senior Applied ML Engineer

Middleton, WI · Remote

$125K - $183K/yr

We are looking for a Senior Applied ML Engineer to be part of revolutionizing these industries. We ... Strong proficiency in Python and ML frameworks (PyTorch, TensorFlow, Hugging Face). * Experience ...

Senior Applied AI Engineer

Middleton, WI · On-site

$127K - $187K/yr

Stay current with emerging AI frameworks, agentic orchestration techniques, and developer tools and ... Highly skilled in Python and at least one major ML framework (PyTorch, TensorFlow). * Knowledge of ...

Senior AI/ML Engineer

Madison, WI · Remote

$90 - $100/hr

Strong Python skills with TensorFlow and PyTorch. * Proven AWS expertise including Bedrock, Lambda ... AWS certification required such as Solutions Architect Associate, Developer Associate, Machine ...

New

Sr Software Engineer

Madison, WI · On-site

$106K - $145K/yr

We are seeking a Senior Software Engineer to join the team building the control system software for ... Experience with Numpy, Scipy, and Pytorch * Experience with image analysis, emphasis on realtime ...

Sr Software Engineer

Madison, WI · On-site

$106K - $145K/yr

We are seeking a Senior Software Engineer to join the team building the control system software for ... Experience with Numpy, Scipy, and Pytorch * Experience with image analysis, emphasis on realtime ...

Sr Software Engineer

Madison, WI · On-site

$106K - $145K/yr

We are seeking a Senior Software Engineer to join the team building the control system software for ... Experience with Numpy, Scipy, and Pytorch * Experience with image analysis, emphasis on realtime ...

Sr Software Engineer

Madison, WI · On-site

$106K - $145K/yr

We are seeking a Senior Software Engineer to join the team building the control system software for ... Experience with Numpy, Scipy, and Pytorch * Experience with image analysis, emphasis on realtime ...

Sr. BI/DW Engineer

Madison, WI · On-site

$90K - $120K/yr

Strong experience with machine learning frameworks and libraries (e.g., TensorFlow, PyTorch, Scikit ... Proven experience as a BI Developer or in a similar data-focused role. * Expert in BI tools mainly ...

New

AI Agent ML Engineer

Madison, WI · On-site

$165K - $190K/yr

Strong programming skills in Python; experience with ML frameworks (PyTorch, TensorFlow) and agent orchestration tools. * Experience in business process analysis, process mapping, and workflow ...

AI Agent ML Engineer

Madison, WI · On-site

$165K - $190K/yr

Strong programming skills in Python; experience with ML frameworks (PyTorch, TensorFlow) and agent orchestration tools. * Experience in business process analysis, process mapping, and workflow ...

next page

Showing results 1-20

Pytorch Developer information

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

Infographic showing various Pytorch Developer job openings in Madison, WI as of May 2026, with employment types broken down into 86% Full Time, 2% Part Time, and 12% Contract. Highlights an 80% Physical, 5% Hybrid, and 15% Remote job distribution.
AI & Machine Learning Developer

AI & Machine Learning Developer

Johnson Health Tech North America Inc

Cottage Grove, WI • On-site

Other

Medical, Dental, Life, Retirement, PTO

Posted 25 days ago


Johnson Health Tech rating

8.1

Company rating: 8.1 out of 10

Based on 5 frontline employees who took The Breakroom Quiz

119th of 416 rated machine equipment manufacturers


Job description

Description


Position Overview:

Under the direction of the Sr. Director of Electrical Engineering, the AI/ML Developer - Mobile Fitness Applications will The AI/ML Developer will design and prototype advanced artificial intelligence features for Johnson Health Tech's mobile fitness applications. This role focuses on leveraging Large Language Models (LLMs) and AWS backend services to create innovative, personalized user experiences. The developer will collaborate closely with client-side Android developers to integrate these features into production applications.


Responsibilities:

Research, design, and implement AI/ML solutions for mobile fitness applications.

Develop and fine-tune LLMs for natural language interactions and personalization.

Build scalable backend services using AWS technologies (Lambda, DynamoDB, SageMaker, etc.).

Build scalable production ready ML ops pipeline and inference endpoints using AWS technologies e.g. SageMaker, Bedrock.

Collaborate with Android developers to integrate AI features into client-side applications.

Create prototypes and proof-of-concepts for new AI-driven features.

Stay current with emerging AI/ML technologies and best practices.

Ensure compliance with data privacy and security standards.

Requirements


Education:

Bachelor's or Master's degree in Computer Science, Data Science, or a related field.


Experience:

3+ years of experience in AI/ML development, with a focus on Natural Language Processing (NLP) and LLMs.

Hands-on experience with AWS services for AI/ML deployment.

Proficiency in Python and ML frameworks (TensorFlow, PyTorch).

Experience with RESTful APIs and microservices architecture.


Benefits:

We offer an excellent compensation package and team-oriented work environment with growth opportunities. Some of our outstanding benefits include:

Health & Dental Insurance

Company paid Life Insurance

401(k)

Paid Time Off benefits

Product discounts

Wellness programs



 Equal Opportunity Employer, including Veterans and Individuals with Disabilities