1

Pytorch Jobs in Pittsburgh, PA (NOW HIRING)

Machine Learning Systems Engineer

Pittsburgh, PA ยท On-site +1

$144K - $192K/yr

Utilize profiling tools (e.g., Nsight, PyTorch Profiler) to identify bottlenecks in data loading, gradient computation, and communication. Implement optimizations like kernel fusion, sharding, and ...

Utilize profiling tools (e.g., Nsight, PyTorch Profiler) to identify bottlenecks in data loading, gradient computation, and communication. Implement optimizations like kernel fusion, sharding, and ...

Proficiency in Python and experience with frameworks like PyTorch, TensorFlow, SciKit, etc. * Basic understanding of Retrieval-Augmented Generation (RAG) and vector databases. * Comfortable working ...

Must-Have Skills 3+ years of ML engineering experience -- model training, fine-tuning, or post-training pipelines in research or production Strong Python and deep learning proficiency (PyTorch ...

Engage in coding, data preparation, and model training using tools such as Python, Azure, TensorFlow, and PyTorch-while remaining open to exploring new technologies. Solve Complex Business Challenges:

next page

Showing results 1-20

Pytorch information

See Pittsburgh, PA salary details

$76K

$136.2K

$195.4K

How much do pytorch jobs pay per year?

As of Jul 30, 2026, the average yearly pay for pytorch in Pittsburgh, PA is $136,234.00, according to ZipRecruiter salary data. Most workers in this role earn between $97,229.00 and $166,611.00 per year, depending on experience, location, and employer.

What are the key skills and qualifications needed to thrive in the Pytorch position, and why are they important?

To thrive in a PyTorch developer role, you need a strong background in deep learning, programming (especially Python), and a solid understanding of machine learning fundamentals, often supported by a degree in computer science, engineering, or a related field. Experience with PyTorch, CUDA, cloud platforms (like AWS or Azure), and familiarity with data processing pipelines are highly valued, and certifications in AI or machine learning can be beneficial. Key soft skills include problem-solving, teamwork, and effective communication to collaborate with cross-functional teams and present technical results clearly. These skills are crucial for building robust machine learning models, ensuring reproducibility, and driving innovation in fast-paced, data-driven environments.

What kinds of projects or tasks can a PyTorch developer expect to work on in a typical role?

As a PyTorch developer, you will likely work on developing, refining, and deploying deep learning models for tasks such as image recognition, natural language processing, or recommendation systems, depending on your company's focus. Your responsibilities may include data preprocessing, model architecture design, experimentation, performance tuning, and collaborating with data scientists and software engineers to integrate models into production systems. You might also be called upon to conduct research or prototype new algorithms, keeping up with the latest advancements in the AI field. Projects can vary from quick proofs of concept to large-scale deployments, offering diverse opportunities to grow your technical and collaborative skills.

What is a PyTorch job?

A PyTorch job typically involves working with the PyTorch deep learning framework to develop, train, and deploy machine learning models. Professionals in this role may build neural networks, perform data preprocessing, optimize models, and integrate them into applications. These jobs are commonly found in AI research, software development, and data science, requiring expertise in Python, deep learning, and model optimization techniques.

What job categories do people searching Pytorch jobs in Pittsburgh, PA look for? The top searched job categories for Pytorch jobs in Pittsburgh, PA are:
What cities near Pittsburgh, PA are hiring for Pytorch jobs? Cities near Pittsburgh, PA with the most Pytorch job openings:
Infographic showing various Pytorch job openings in Pittsburgh, PA as of July 2026, with employment types broken down into 1% Internship, 90% Full Time, 7% Part Time, and 2% Contract. Highlights an 85% Physical, 3% Hybrid, and 12% Remote job distribution, with an average salary of $136,234 per year, or $65.5 per hour.

Machine Learning Systems Engineer

Motional

Pittsburgh, PA โ€ข On-site, Remote

Other

Re-posted 19 days ago


Job description

Mission Summary:

We are looking for a Machine Learning Systems Engineer to join our ML Acceleration team. In this role, you will be responsible for the core systems that enable our researchers to train frontier models at scale, focusing obsessively on speed, cost, reliability, and throughput. You will work at the intersection of machine learning research and high-performance systems engineering. Your work will directly impact our ability to scale large-scale distributed model training and reduce the time-to-convergence for our next generation of models.

What you'll be doing:

  • Performance Profiling & Optimization: Utilize profiling tools (e.g., Nsight, PyTorch Profiler) to identify bottlenecks in data loading, gradient computation, and communication. Implement optimizations like kernel fusion, sharding, and tiling to improve step time.
  • Distributed Training: Optimize distributed training pipelines using frameworks such as PyTorch Distributed.
  • Kernel Development: Design and maintain high-performance GPU kernels in Triton or CUDA for state-of-the-art ML workloads.
  • Data Pipeline Engineering: Optimize robust data loading pipelines that maximize training throughput.

What we're looking for:

  • Education: Bachelor's, Master's degree, or PhD in Computer Science, Computer Engineering, or a related technical discipline.
  • Software Engineering: Strong proficiency in Python.
  • ML Frameworks: Extensive hands-on experience with PyTorch.
  • ML Knowledge: Experience optimizing machine learning model execution during training and inference, alongside a strong understanding of fundamental machine learning concepts, architectures, and processes.
  • Problem Solving: Exceptional analytical and problem-solving skills, with a bias for action and a data-driven approach to technical challenges.

We encourage a hybrid schedule with in-office time at one of our locations in Boston, Pittsburgh, or Las Vegas to support collaboration, or this role can be fully remote.