ZipRecruiter

Log In

1

Pytorch Developer Jobs in Madison, WI (NOW HIRING)

Paradigm

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

Bespoke Labs

Machine Learning Engineer

Whitewater, WI · On-site

SFT, RLHF, PPO, DPO, and reward model training in PyTorch * Build and maintain the infrastructure around RL training: rollout collection, data curation, reward model serving, and experiment ...

Bespoke Labs

Machine Learning Engineer

Madison, WI

SFT, RLHF, PPO, DPO, and reward model training in PyTorch * Build and maintain the infrastructure around RL training: rollout collection, data curation, reward model serving, and experiment ...

Bespoke Labs

Machine Learning Engineer

Madison, WI · On-site

SFT, RLHF, PPO, DPO, and reward model training in PyTorch * Build and maintain the infrastructure around RL training: rollout collection, data curation, reward model serving, and experiment ...

All JobsPytorch Developer JobsPytorch Developer Jobs in Madison, WI

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 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 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 popular job titles related to Pytorch Developer jobs in Madison, WI?

For Pytorch Developer jobs in Madison, WI, the most frequently searched job titles are:

What job categories do people searching Pytorch Developer jobs in Madison, WI look for?

The top searched job categories for Pytorch Developer jobs in Madison, WI are:

Pytorch Developer jobs near you

AI & Machine Learning Developer

Cottage Grove, WI • On-site

Johnson Health Tech North America Inc
Wholesale • 1 - 5K employees
Learn more about Johnson Health Tech North America Incexternal

Full-time

Medical, Dental, Life, Retirement, PTO

Re-posted 2 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

156th of 499 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 

What Johnson Health Tech employees say

Pay

Workplace

Get the full story on Breakroom

Apply