PyTorch Jobs

PyTorch is an open-source Python-based machine learning framework used to build, train and deploy neural networks. Engineers use it for deep learning, GPU-accelerated computation, model research and production AI systems.

Browse 201 open PyTorch jobs from 16 companies including Nvidia, Anthropic, and OpenAI. Popular hiring locations include Santa Clara, San Francisco, and United States.

PyTorch Salary Ranges

Pay is quoted per market: the same role is worth several times more in one city than in another, so a single figure spanning countries would describe none of them. 183 of these 201 listings publish a salary. Markets with fewer than 8 disclosed salaries are not shown.

Median yearly pay by country, each in the currency named in its own row

Median yearly pay by metro, each in the currency named in its own row

A job open in more than one place is counted by each market it is open in, so the Listings column adds up to more than the number of listings.

Seniority and Work Setup for PyTorch Roles

A role can be open to more than one level, so the shares add up to more than 100%.

A role can offer more than one arrangement, so the shares add up to more than 100%.

14 offer visa sponsorship and 13 offer relocation assistance.

Skills Commonly Paired with PyTorch

Top Companies Hiring for PyTorch

Employers with the most open PyTorch jobs

Where PyTorch Jobs Are Located

Where these PyTorch jobs are based

201 Open PyTorch Jobs

Staff Machine Learning Engineer, Radar at Stripe SENIOR

📍 United States

💰 USD 253,500-380,300 per year

✅ Remote

AI Data Pipelines @ 4 Data Science Deep Learning @ 4 FinTech @ 4 Fraud @ 4 LLM Machine Learning @ 7 Mathematics @ 6 NLP @ 4 Payments PyTorch @ 6 Spark @ 6 Statistics @ 6 TensorFlow @ 6 XGBoost @ 6

Machine Learning Engineer, Radar at Stripe MIDDLE

📍 South San Francisco, United States
📍 New York City, United States
📍 Seattle, United States

💰 USD 180,000-270,000 per year

✅ On-site

AI Data Analysis @ 6 Deep Learning Fraud @ 3 Machine Learning @ 3 Payments PyTorch @ 5 Python @ 5 SQL @ 5 Spark @ 5 Statistics @ 6

Machine Learning Engineer (AI/ML) at ClickHouse SENIOR

📍 Phoenix, United States
📍 Detroit, United States
📍 Canada
📍 Austin, United States
📍 Dallas, United States
📍 Los Angeles, United States
📍 New York City, United States
📍 San Francisco, United States

💰 USD 150,000-238,000 per year

✅ Remote

AI @ 4 API @ 4 AWS @ 3 Azure @ 3 ClickHouse @ 1 GCP @ 3 Machine Learning @ 4 Observability PyTorch @ 4 Python @ 4 Slurm @ 4 Vector Databases @ 4

PhD Research Intern, Cross-Disciplinary Vision Science – Summer 2027 at Nvidia INTERN

📍 Santa Clara, United States

💸 USD 38-94 per hour

✅ On-site

AI @ 3 Computer Vision Deep Learning @ 2 GPU @ 3 GitHub @ 3 LLM PyTorch @ 3 Python @ 5 Rust @ 5 Software Development @ 3

Senior Machine Learning Engineer, Perception - Autonomous Driving at Nvidia SENIOR

📍 Santa Clara, United States

💰 USD 184,000-356,500 per year

✅ On-site

AI Algorithms @ 7 Communication @ 6 Computer Vision Data Pipelines Debugging @ 7 Deep Learning @ 7 Experimentation GPU @ 7 Machine Learning @ 7 PyTorch @ 4 Python @ 7 Robotics @ 4 TensorFlow @ 4 TensorRT @ 4

Performance Engineer, Deep Learning and HPC at Nvidia MIDDLE

📍 Santa Clara, United States

💰 USD 116,000-212,800 per year

✅ On-site

AWS @ 3 Automated Testing Azure @ 3 CentOS @ 6 Communication @ 3 Data Analysis @ 6 Debugging @ 5 Deep Learning @ 3 Docker @ 2 GCP @ 3 GPU @ 3 HPC @ 3 Linux @ 6 Machine Learning Marketing Performance Analysis PyTorch @ 3 Python @ 3 TensorRT @ 3

NVIDIA 2027 Ignite Internships: Software Engineering at Nvidia INTERN

📍 Santa Clara, United States

💸 USD 20-56 per hour

✅ On-site

AI Android Computer Vision GPU Leadership @ 3 Linux @ 3 Machine Learning Mathematics @ 3 Matlab @ 3 Perl @ 3 PyTorch @ 3 Python @ 3 Robotics Security

Performance Engineer - Deep Learning at Nvidia MIDDLE SENIOR

📍 Santa Clara, United States

💰 USD 124,000-241,500 per year

✅ On-site

AI @ 6 CUDA @ 3 Communication @ 3 Deep Learning @ 6 GPU @ 3 Hiring @ 6 JAX @ 3 LLM @ 3 Performance Analysis @ 3 Profiling @ 3 PyTorch @ 3 Python @ 3

Senior Deep Learning Scientist, Multimodal Agentic RL at Nvidia SENIOR

📍 Santa Clara, United States

💰 USD 152,000-287,500 per year

✅ On-site

AI @ 4 Agentic Systems @ 4 Algorithms @ 4 Deep Learning @ 7 LLM Machine Learning @ 7 Mathematics @ 4 PyTorch @ 7 Python @ 7 Reinforcement Learning @ 4

Senior Deep Learning Scientist, Multimodal Agentic RL at Nvidia SENIOR

📍 Santa Clara, United States

💰 USD 184,000-287,500 per year

✅ On-site

AI @ 4 Agentic Systems @ 4 Algorithms @ 4 Deep Learning @ 7 LLM Machine Learning @ 7 Mathematics @ 7 PyTorch @ 7 Python @ 7 Reinforcement Learning @ 4

Senior Production Engineer - DGX Cloud at Nvidia SENIOR

📍 Santa Clara, United States

💰 USD 184,000-356,500 per year

✅ On-site

AI @ 4 AWS Azure CI/CD @ 4 CUDA @ 3 Communication @ 6 Debugging @ 6 Distributed Systems @ 4 GPU @ 3 Go @ 7 IaC Kubernetes @ 7 LLM @ 3 Linux @ 7 NCCL @ 3 Networking @ 7 Observability Performance Analysis @ 3 PyTorch @ 3 Python @ 7 SGLang @ 3 SRE @ 4 Security TensorRT @ 3 Terraform @ 4 vLLM @ 3

PhD Research Intern, Embodied and Agentic AI - 2027 at Nvidia INTERN

📍 Santa Clara, United States

💸 USD 38-94 per hour

✅ On-site

AI @ 7 Agentic AI CUDA @ 3 Communication @ 3 Computer Vision @ 7 Experimentation LLM Machine Learning @ 3 Parallel Programming @ 3 PyTorch @ 3 Python @ 3

Generation time: 0.291