PyTorch with Triton performance Engineer Bellevue, WA (Onsite) Contract / FTE Job Summary Design and implement high intensity stress workloads using PyTorch and Triton to identify performance ...
PyTorch with Triton performance Engineer Bellevue, WA (Onsite) Contract / FTE Job Summary Design and implement high intensity stress workloads using PyTorch and Triton to identify performance ...
Compiler Engineer - MLIR / PyTorch Infrastructure
Palo Alto, CA · Remote
$110K - $144K/yr
Working closely with hardware engineers and ML developers, your work will expand interoperability with PyTorch and other frameworks, laying the groundwork for long-term innovation. The result: an ...
Quick apply
Compiler Engineer - MLIR / PyTorch Infrastructure
Palo Alto, CA · Remote
$110K - $144K/yr
Working closely with hardware engineers and ML developers, your work will expand interoperability with PyTorch and other frameworks, laying the groundwork for long-term innovation. The result: an ...
Compiler Engineer - MLIR / PyTorch Infrastructure
Palo Alto, CA · On-site
$126K - $165K/yr
Working closely with hardware engineers and ML developers, your work will expand interoperability with PyTorch and other frameworks, laying the groundwork for long-term innovation. The result: an ...
Compiler Engineer - MLIR / PyTorch Infrastructure
Palo Alto, CA · On-site
$126K - $165K/yr
Working closely with hardware engineers and ML developers, your work will expand interoperability with PyTorch and other frameworks, laying the groundwork for long-term innovation. The result: an ...
Compiler Engineer - MLIR / PyTorch Infrastructure
Austin, TX · On-site
$106K - $139K/yr
Working closely with hardware engineers and ML developers, your work will expand interoperability with PyTorch and other frameworks, laying the groundwork for long-term innovation. The result: an ...
Compiler Engineer - MLIR / PyTorch Infrastructure
Austin, TX · On-site
$106K - $139K/yr
Working closely with hardware engineers and ML developers, your work will expand interoperability with PyTorch and other frameworks, laying the groundwork for long-term innovation. The result: an ...
Compiler Engineer - MLIR / PyTorch Infrastructure
$126K - $165K/yr
Working closely with hardware engineers and ML developers, your work will expand interoperability with PyTorch and other frameworks, laying the groundwork for long-term innovation. The result: an ...
Compiler Engineer - MLIR / PyTorch Infrastructure
$126K - $165K/yr
Working closely with hardware engineers and ML developers, your work will expand interoperability with PyTorch and other frameworks, laying the groundwork for long-term innovation. The result: an ...
Compiler Engineer - MLIR / PyTorch Infrastructure
Austin, TX · Remote
$110K - $144K/yr
Working closely with hardware engineers and ML developers, your work will expand interoperability with PyTorch and other frameworks, laying the groundwork for long-term innovation. The result: an ...
Quick apply
Compiler Engineer - MLIR / PyTorch Infrastructure
Austin, TX · Remote
$110K - $144K/yr
Working closely with hardware engineers and ML developers, your work will expand interoperability with PyTorch and other frameworks, laying the groundwork for long-term innovation. The result: an ...
PRINCIPAL SOFTWARE DEVELOPMENT ENGINEER - PYTORCH TRAINING FRAMEWORKS THE ROLE: AMD is looking for a Principal-level PyTorch training framework expert to help drive performance, scalability, and ...
PRINCIPAL SOFTWARE DEVELOPMENT ENGINEER - PYTORCH TRAINING FRAMEWORKS THE ROLE: AMD is looking for a Principal-level PyTorch training framework expert to help drive performance, scalability, and ...
Compiler Engineer - MLIR / PyTorch Infrastructure
$106K - $139K/yr
Working closely with hardware engineers and ML developers, your work will expand interoperability with PyTorch and other frameworks, laying the groundwork for long-term innovation. The result: an ...
Compiler Engineer - MLIR / PyTorch Infrastructure
$106K - $139K/yr
Working closely with hardware engineers and ML developers, your work will expand interoperability with PyTorch and other frameworks, laying the groundwork for long-term innovation. The result: an ...
Principal Software Engineer - PyTorch Training Frameworks
San Jose, CA · On-site
$210K/yr
PRINCIPAL SOFTWARE DEVELOPMENT ENGINEER - PYTORCH TRAINING FRAMEWORKS THE ROLE: AMD is looking for a Principal-level PyTorch training framework expert to help drive performance, scalability, and ...
Principal Software Engineer - PyTorch Training Frameworks
San Jose, CA · On-site
$210K/yr
PRINCIPAL SOFTWARE DEVELOPMENT ENGINEER - PYTORCH TRAINING FRAMEWORKS THE ROLE: AMD is looking for a Principal-level PyTorch training framework expert to help drive performance, scalability, and ...
We are currently hiring a Senior Python Engineer ( Pytorch, CI/CD experience) for a remote role in US. Essential functions * Resolving merge conflicts in version control systems (like git or ...
We are currently hiring a Senior Python Engineer ( Pytorch, CI/CD experience) for a remote role in US. Essential functions * Resolving merge conflicts in version control systems (like git or ...
MLOps Engineer - JAX/PyTorch Expert
New York, NY · Remote
$90 - $130/hr
MLOps Engineer (JAX, PyTorch, Pallas/Triton) Type: Contract Compensation: $90-$130/hour Location: Remote Role Responsibilities * Guide research and engineering teams to close knowledge gaps and ...
Quick apply
MLOps Engineer - JAX/PyTorch Expert
New York, NY · Remote
$90 - $130/hr
MLOps Engineer (JAX, PyTorch, Pallas/Triton) Type: Contract Compensation: $90-$130/hour Location: Remote Role Responsibilities * Guide research and engineering teams to close knowledge gaps and ...
Senior Python AI/ML Engineer (TensorFlow/PyTorch) Location: Alpharetta, GA (willing to travel to client locations) Employment Type: Full-Time (W2) Role Overview We are seeking an accomplished Senior ...
Senior Python AI/ML Engineer (TensorFlow/PyTorch) Location: Alpharetta, GA (willing to travel to client locations) Employment Type: Full-Time (W2) Role Overview We are seeking an accomplished Senior ...
... PyTorch. • Experience with acoustic and linguistic feature extraction and understanding ofspeaker identification and verification systems. • Familiarity with natural language processing (NLP) and ...
... PyTorch. • Experience with acoustic and linguistic feature extraction and understanding ofspeaker identification and verification systems. • Familiarity with natural language processing (NLP) and ...
Staff Engineer, Compiler
San Jose, CA · On-site
You'll engage with upstream review processes for PyTorch, Triton, Helion, and adjacent projects. What You Bring * Bachelor's with 10+ years, or Master's with 8+ years, or PhD's with 5+ years of ...
Staff Engineer, Compiler
San Jose, CA · On-site
You'll engage with upstream review processes for PyTorch, Triton, Helion, and adjacent projects. What You Bring * Bachelor's with 10+ years, or Master's with 8+ years, or PhD's with 5+ years of ...
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow...
New York, NY · On-site
$112K - $147K/yr
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow on AWS) As a Capital One Machine Learning Engineer (MLE) , you'll be part of an Agile team dedicated ...
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow...
New York, NY · On-site
$112K - $147K/yr
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow on AWS) As a Capital One Machine Learning Engineer (MLE) , you'll be part of an Agile team dedicated ...
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow...
Mclean, VA · On-site
$103K - $136K/yr
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow on AWS) As a Capital One Machine Learning Engineer (MLE) , you'll be part of an Agile team dedicated ...
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow...
Mclean, VA · On-site
$103K - $136K/yr
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow on AWS) As a Capital One Machine Learning Engineer (MLE) , you'll be part of an Agile team dedicated ...
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow...
Cambridge, MA · On-site
$112K - $147K/yr
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow on AWS) As a Capital One Machine Learning Engineer (MLE) , you'll be part of an Agile team dedicated ...
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow...
Cambridge, MA · On-site
$112K - $147K/yr
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow on AWS) As a Capital One Machine Learning Engineer (MLE) , you'll be part of an Agile team dedicated ...
Senior Data Scientist Consultant (Python, SQL, TensorFlow, PyTorch, Scikit-Learn, FTE, Onsite)
Plano, TX · On-site +1
$63K - $131K/yr
We are currently seeking a Senior Data Scientist Consultant (Python, SQL, TensorFlow, PyTorch, Scikit-Learn, FTE, Onsite) to join our team in Plano , Texas (US-TX) , United States (US) . "Please note ...
Senior Data Scientist Consultant (Python, SQL, TensorFlow, PyTorch, Scikit-Learn, FTE, Onsite)
Plano, TX · On-site +1
$63K - $131K/yr
We are currently seeking a Senior Data Scientist Consultant (Python, SQL, TensorFlow, PyTorch, Scikit-Learn, FTE, Onsite) to join our team in Plano , Texas (US-TX) , United States (US) . "Please note ...
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow...
San Jose, CA · On-site
$120K - $158K/yr
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow on AWS) As a Capital One Machine Learning Engineer (MLE) , you'll be part of an Agile team dedicated ...
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow...
San Jose, CA · On-site
$120K - $158K/yr
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow on AWS) As a Capital One Machine Learning Engineer (MLE) , you'll be part of an Agile team dedicated ...
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow...
Mclean, VA · On-site
$103K - $136K/yr
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow on AWS) As a Capital One Machine Learning Engineer (MLE) , you'll be part of an Agile team dedicated ...
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow...
Mclean, VA · On-site
$103K - $136K/yr
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow on AWS) As a Capital One Machine Learning Engineer (MLE) , you'll be part of an Agile team dedicated ...
Pytorch information
See salary details
$80.5K - $92K
11% of jobs
$102.5K is the 25th percentile. Wages below this are outliers.
$92K - $103.5K
16% of jobs
$103.5K - $115K
16% of jobs
$115K - $126.5K
5% of jobs
$126.5K - $138K
0% of jobs
$138K - $149.5K
0% of jobs
$149.5K - $161K
1% of jobs
The median wage is $161.8K / yr.
$161K - $172.5K
22% of jobs
$175.8K is the 75th percentile. Wages above this are outliers.
$172.5K - $184K
16% of jobs
$184K - $195.5K
3% of jobs
$195.5K - $207K
11% of jobs
$80.5K
$144.3K
$207K
How much do pytorch jobs pay per year?
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.

Other
Posted yesterday
Job description
PyTorch with Triton performance Engineer
Bellevue, WA (Onsite)
Contract / FTE
Job Summary
Design and implement high intensity stress workloads using PyTorch and Triton to identify performance bottlenecks and improve platform stability and maturity
Job Description
Design and implement high intensity stress workloads using PyTorch and Triton Exercise core MAIA execution paths including compute memory DMA and collectives Enable early detection of performance cliffs stability issues and system bottlenecks across simulator and real hardware Improve platform maturity reduce latestage escapes and increase confidence for broader internal and external adoption Develop PyTorch workloads stressing modellevel execution such as large GEMMs attention patterns MoElike behavior mixed precision and longrunning loops Author custom Triton kernels to stress hardware execution units memory hierarchies and synchronization paths Build parameterized stress harnesses scalable by problem size number of devices and runtime duration Integrate workloads with existing profiling monitoring and failure triage tooling Collaborate with platform firmware and SDK teams to target known risk areas and emerging issues Document usage patterns and provide reproducible scripts for lab and continuous integration CI usage
Roles and Responsibilities : Develop and maintain a library of reusable PyTorch stress workloads Create Tritonbased micro and macrokernels designed specifically for stress and saturation testing Build and support test harnesses and scripts for singledevice and multidevice execution Ensure workload designs align with platform risk areas and emerging hardwaresoftware issues Collaborate crossfunctionally with platform firmware and SDK teams to refine stress tests Provide comprehensive documentation describing workload intent configuration options and expected stress characteristics Support profiling monitoring and failure triage by integrating stress workloads with existing tools Deliver reproducible and scalable testing solutions for lab and CI environments
About V Dart
Sourced by ZipRecruiter
Company size
1,001 - 5,000 Employees
Headquarters location
Alpharetta, GA, US
Year founded
2008