Senior GPU Software Engineer - PyTorch / Triton (Kernel Development) Location: Remote Role Summary We are seeking expert-level GPU software engineers to support a high-visibility platform initiative ...
Senior GPU Software Engineer - PyTorch / Triton (Kernel Development) Location: Remote Role Summary We are seeking expert-level GPU software engineers to support a high-visibility platform initiative ...
Machine Learning Systems Engineer
Las Vegas, NV · On-site +1
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 ...
Machine Learning Systems Engineer
Las Vegas, NV · On-site +1
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 ...
Senior Machine Learning Engineer - SIML, ISE
Cupertino, CA · On-site
$128K - $177K/yr
We are especially looking for PyTorch-focused ML experts driving system-level efficiency from on-device to large-scale models. If you have deep experience with PyTorch internals and high-performance ...
Senior Machine Learning Engineer - SIML, ISE
Cupertino, CA · On-site
$128K - $177K/yr
We are especially looking for PyTorch-focused ML experts driving system-level efficiency from on-device to large-scale models. If you have deep experience with PyTorch internals and high-performance ...
Agentic AI Developer
Jersey City, NJ · On-site
AI enabled chatbot Langchain TensorFlow Java PyTorch RASA Python AI Experience: 10-14 yrs Required Skills: • Proven experience in developing AI agents, chatbots, or conversational AI systems. • ...
Quick apply
Agentic AI Developer
Jersey City, NJ · On-site
AI enabled chatbot Langchain TensorFlow Java PyTorch RASA Python AI Experience: 10-14 yrs Required Skills: • Proven experience in developing AI agents, chatbots, or conversational AI systems. • ...
Senior Machine Learning Engineer - Foundation Model
Santa Clara, CA · On-site
$122K - $168K/yr
PyTorch Lightning) • Knowledge of transformer architecture and ways to accelerate the training and inference of transformer models. • Experience of using pytorch ddp for distributed training of ...
Senior Machine Learning Engineer - Foundation Model
Santa Clara, CA · On-site
$122K - $168K/yr
PyTorch Lightning) • Knowledge of transformer architecture and ways to accelerate the training and inference of transformer models. • Experience of using pytorch ddp for distributed training of ...
Machine Learning Systems Engineer
Pittsburgh, PA · On-site +1
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 ...
Machine Learning Systems Engineer
Pittsburgh, PA · On-site +1
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 ...
Machine learning Engineer
Pennington, NJ · On-site
$105K - $125K/yr
Strong Python (NumPy, pandas, scikit-learn, PyTorch/TensorFlow) Experience on AWS/Azure/GCP ML services Roles & Responsibilities Looking for a strong hands-on ML Engineer with deep experience in data ...
Machine learning Engineer
Pennington, NJ · On-site
$105K - $125K/yr
Strong Python (NumPy, pandas, scikit-learn, PyTorch/TensorFlow) Experience on AWS/Azure/GCP ML services Roles & Responsibilities Looking for a strong hands-on ML Engineer with deep experience in data ...
Machine Learning Systems Engineer
Boston, MA · On-site +1
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 ...
Machine Learning Systems Engineer
Boston, MA · On-site +1
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 ...
Senior Life Cycle Engineer, PyTorch - Deep Learning
Santa Clara, CA · On-site
$122K - $168K/yr
Are you passionate about building software for large-scale distributed projects? NVIDIA's Deep Learning Frameworks Teams seek Senior Software Engineers to create systems for continuous integration ...
Senior Life Cycle Engineer, PyTorch - Deep Learning
Santa Clara, CA · On-site
$122K - $168K/yr
Are you passionate about building software for large-scale distributed projects? NVIDIA's Deep Learning Frameworks Teams seek Senior Software Engineers to create systems for continuous integration ...
Develop and deploy ML/LLM solutions using Python and PyTorch * Integrate AI capabilities into applications and services * Evaluate and implement LLMs (OpenAI, Anthropic, Google, Meta, etc.) based on ...
Develop and deploy ML/LLM solutions using Python and PyTorch * Integrate AI capabilities into applications and services * Evaluate and implement LLMs (OpenAI, Anthropic, Google, Meta, etc.) based on ...
Data Scientist
Plano, TX · On-site
Apply machine learning algorithms using libraries such as Scikit-learn, TensorFlow, or PyTorch . * Evaluate model performance and tune hyperparameters for improved accuracy. Visualization & Reporting
Data Scientist
Plano, TX · On-site
Apply machine learning algorithms using libraries such as Scikit-learn, TensorFlow, or PyTorch . * Evaluate model performance and tune hyperparameters for improved accuracy. Visualization & Reporting
Machine Learning Specialist | Remote
Manhattan, NY · Remote
$70 - $250/hr
... such as PyTorch or TensorFlow Support model deployment and MLOps workflows Participate in AI-driven opportunity matching for full-time, part-time, or project-based roles Complete an initial AI ...
Machine Learning Specialist | Remote
Manhattan, NY · Remote
$70 - $250/hr
... such as PyTorch or TensorFlow Support model deployment and MLOps workflows Participate in AI-driven opportunity matching for full-time, part-time, or project-based roles Complete an initial AI ...
ML Engineer
Minneapolis, MN · On-site
Python, TensorFlow, PyTorch, Docker, REST APIs
ML Engineer
Minneapolis, MN · On-site
Python, TensorFlow, PyTorch, Docker, REST APIs
Sr. Java Developer with strong AI/ML exp
Lake Mary, FL · On-site
$50 - $63.75/hr
Python NumPy, Pandas, TensorFlow/PyTorch. * Java Spring Boot, multithreading, JVM optimization. Required Skills: * 10 or more years of experience in data science. * Ability to work effectively in a ...
Quick apply
Sr. Java Developer with strong AI/ML exp
Lake Mary, FL · On-site
$50 - $63.75/hr
Python NumPy, Pandas, TensorFlow/PyTorch. * Java Spring Boot, multithreading, JVM optimization. Required Skills: * 10 or more years of experience in data science. * Ability to work effectively in a ...
Python Automation Developer (Onsite)
Seattle, WA · On-site
$57.25 - $78.75/hr
TensorFlow and PyTorch; 4. Framework: Django ; Flask; FastAPI 5. Web Scraping and HTTP: Beautiful Soup; Scrapy Must Have: 3-4 Years of Experience in developing Automations using Python language Core ...
Python Automation Developer (Onsite)
Seattle, WA · On-site
$57.25 - $78.75/hr
TensorFlow and PyTorch; 4. Framework: Django ; Flask; FastAPI 5. Web Scraping and HTTP: Beautiful Soup; Scrapy Must Have: 3-4 Years of Experience in developing Automations using Python language Core ...
ML/AI Engineers
Santa Clara, CA · On-site
Python (Advanced), Machine Learning frameworks (TensorFlow/PyTorch) Mandatory: Kubernetes experience and cloud-native development practices Required Skills & Experience • Primary Technologies:
Quick apply
ML/AI Engineers
Santa Clara, CA · On-site
Python (Advanced), Machine Learning frameworks (TensorFlow/PyTorch) Mandatory: Kubernetes experience and cloud-native development practices Required Skills & Experience • Primary Technologies:
Dataflow, Python, PyTorch, Docker, AWS, Kubernetes, Data Science, Machine Learning/Artificial Intelligence, Git, Continuous Integration, DevOps. About SYSTOLIC: SYSTOLIC is dedicated to giving our ...
Dataflow, Python, PyTorch, Docker, AWS, Kubernetes, Data Science, Machine Learning/Artificial Intelligence, Git, Continuous Integration, DevOps. About SYSTOLIC: SYSTOLIC is dedicated to giving our ...
Senior Software Engineer, AI Networking
$139K - $183K/yr
Develop and deploy a scalable, reliable data curation pipeline capable of handling complex data types, such as time series and PyTorch model graphs, to effectively support the training of high ...
Senior Software Engineer, AI Networking
$139K - $183K/yr
Develop and deploy a scalable, reliable data curation pipeline capable of handling complex data types, such as time series and PyTorch model graphs, to effectively support the training of high ...
Data AI/ML Engineer
Sunnyvale, CA · On-site
Python, TensorFlow, PyTorch, scikit-learn * ML pipelines and data processing systems. * ML models What you will bring: * Expert-level proficiency in Python is standard * In-depth knowledge of ML ...
Data AI/ML Engineer
Sunnyvale, CA · On-site
Python, TensorFlow, PyTorch, scikit-learn * ML pipelines and data processing systems. * ML models What you will bring: * Expert-level proficiency in Python is standard * In-depth knowledge of ML ...
Lead Data Scientist (Scientific Software Engineer / Computational Scientist) - Only W2
Mountain View, CA · On-site
C++ * PyTorch * Geostatistics * 3D Mathematics * CUDA/OpenMP * AI-assisted coding Short Overview: * Scientific Software Engineer or Computational Scientist with a niche background in scientific ...
Quick apply
Lead Data Scientist (Scientific Software Engineer / Computational Scientist) - Only W2
Mountain View, CA · On-site
C++ * PyTorch * Geostatistics * 3D Mathematics * CUDA/OpenMP * AI-assisted coding Short Overview: * Scientific Software Engineer or Computational Scientist with a niche background in scientific ...
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.

Full-time
Posted 23 days ago
Job description
Location: Remote
Job Description:
Role Summary
We are seeking expert-level GPU software engineers to support a high-visibility platform initiative within the Maya program, focused on building software tooling on top of a custom compiler and SDK.
The role involves developing, optimizing, and porting GPU kernels and AI workloads to a specialized hardware platform.
This is a critical and time-sensitive engagement with immediate onboarding expectations and long-term roadmap alignment (~18 months).
Key Responsibilities
Develop GPU kernels for specialized hardware platforms using PyTorch/Triton frameworks
Build software solutions leveraging custom compiler and SDK capabilities
Design and implement kernel-level optimizations to control hardware execution behavior
Port open-source AI/ML models to custom SDK environments
Develop stress testing and validation workloads aligned to hardware behaviour and platform validation
Support testing and stress testing of current and next-generation hardware platforms
Collaborate closely with platform architects and compiler teams to enhance system capabilities
Core Technical Skills (Must-Have)
Programming & Frameworks
Python
C/C++ (systems-level programming)
PyTorch
Triton (Triton language / kernel development)
GPU & Systems Expertise
GPU kernel development (mandatory and critical)
Strong understanding of GPU architecture and compute optimization
Experience with compiler-based optimizations / runtime execution layers
Experience with custom SDKs or hardware abstraction layers
Performance & Workloads
Experience in:
GEMM kernel development (matrix multiplication kernels)
Porting ML models to new hardware platforms
Performance tuning and stress testing at system level