PyTorch AI Jobs

Discover the latest remote and onsite PyTorch AI roles across top active AI companies. Updated hourly.

Check out 362 new PyTorch AI roles opportunities posted on AI Chopping Block

TPM Manager

New
Top rated
Labelbox
Full-time
Full-time
Posted

As an Applied Research Engineer at Labelbox, you will create frameworks and tools to construct, train, benchmark, and evaluate autonomous agent capabilities. You will design agent-focused data programs using supervised fine-tuning (SFT) and reinforcement learning (RL) methodologies, develop data pipelines from diverse sources such as code repositories, web browsers, and computer systems, and implement and adapt popular open-source agent libraries and benchmarks with proprietary datasets and models. Your role includes engaging with research teams in frontier AI labs and the wider AI community to understand evolving agent data needs for frontier models and share best practices, collaborating closely with frontier AI lab customers to understand requirements and guide model development, and publishing research findings in academic journals, conferences, and blog posts.

$250,000 – $300,000
Undisclosed
YEAR

(USD)

San Francisco or Wrocław, United States or Poland
Maybe global
Hybrid
Python
PyTorch
TensorFlow
JAX
Prompt Engineering

Forward Deployed Engineering Manager

New
Top rated
Labelbox
Full-time
Full-time
Posted

As an Applied Research Engineer at Labelbox, you are responsible for creating frameworks and tools to construct, train, benchmark, and evaluate autonomous agent capabilities. You design agent-focused data programs using supervised fine-tuning (SFT) and reinforcement learning (RL) methodologies. You develop data pipelines from diverse sources such as code repositories, web browsers, and computer systems. You implement and adapt popular open-source agent libraries and benchmarks with proprietary datasets and models. You engage with research teams in frontier AI labs and the wider AI community to understand evolving agent data needs and share best practices. You collaborate closely with frontier AI lab customers to understand their requirements and guide model development. Additionally, you publish research findings in academic journals, conferences, and blog posts.

$250,000 – $300,000
Undisclosed
YEAR

(USD)

San Francisco or Wrocław, United States or Poland
Maybe global
Hybrid
Python
PyTorch
JAX
TensorFlow
NLP

Senior Product Engineer, Growth & Lifecycle Infrastructure - Music & Audio

New
Top rated
Stability AI
Full-time
Full-time
Posted

Lead efforts to drive the design and development of customer-facing multi-modal machine learning inference systems. Work with the Platform and Inference teams on building inference systems for the next generation of models, focusing on optimization, model tuning, and deployment. Partner with leading cloud providers to deliver hosted Stability AI inference solutions. Serve as a strategic thought partner for leaders across the organization on driving business impact through machine learning. Contribute to bringing new Stability models and pipelines into existence. Prototype and productionize inference platform improvements and new features.

Undisclosed

()

Los Angeles, United States
Maybe global
Hybrid
Python
PyTorch
Docker
Kubernetes
AWS

Researcher: Agent Post-Training, API & Power-Users

New
Top rated
OpenAI
Full-time
Full-time
Posted

The role involves improving the capabilities, reliability, and product fit of OpenAI’s agentic models for power users and API developers. Responsibilities include designing and running experiments to enhance model behavior in API and power-user workflows such as function calling, tool use, coding, planning, and long-horizon execution. The role requires building evals, graders, and environments from real developer and power-user workflows, turning observed failures into training data, hypotheses, and improvements. The researcher partners with API and power-users to identify behavior gaps and translate product signals into post-training interventions. They improve model behavior when composed into systems, ensuring reliable tool use, respect for developer intent, appropriate error handling, clarification when needed, and task coherence. The role also includes owning end-to-end model behavior projects from failure analysis through training, eval design, integration into major model runs, and launch readiness. Developing feedback loops using power-user traces and production-like environments to identify model failures and gaps is part of the job. The researcher assists in deciding which capabilities, fixes, and integrations are ready for major model runs. Additionally, debugging hard failures in models by analyzing traces, evals, training data, and product context is required. The role involves working on early-training and alignment interventions, improving large-scale training and launch machinery, and taking on cross-functional projects that touch model training, product infrastructure, and production agent harnesses, including multi-agent systems and training against production-like environments.

$295,000 – $445,000
Undisclosed
YEAR

(USD)

San Francisco, United States
Maybe global
Remote
Python
PyTorch
TensorFlow
MLflow
MLOps

Senior Deep Learning Engineer (음성 합성 개발)

New
Top rated
42dot
Full-time
Full-time
Posted

Research and develop latest TTS models based on LLM and Flow Matching; develop and advance emotion controllable TTS models; build and improve quality of speech synthesis data using latest generative models; develop and apply multilingual and multi-speaker TTS models to services; optimize TTS models for server and on-device environments; develop real-time (streaming) speech synthesis systems and optimize latency; improve inference and training pipelines to enhance speech generation quality.

Undisclosed

()

Pangyo, South Korea
Maybe global
Remote
Python
C++
PyTorch
TensorFlow
Model Evaluation

Researcher, Training - London

New
Top rated
OpenAI
Full-time
Full-time
Posted

Design, prototype and scale up new architectures to improve model intelligence; execute and analyze experiments autonomously and collaboratively; study, debug, and optimize both model performance and computational performance; contribute to training and inference infrastructure.

£170,000 – £445,000
Undisclosed
YEAR

(GBP)

London, United Kingdom
Maybe global
Hybrid
Python
PyTorch
TensorFlow
Transformers
Model Evaluation

Software Engineer, Computer Vision and Deep Learning

New
Top rated
Mashgin
Full-time
Full-time
Posted

Developing new computer vision algorithms with founders in C/C++ and Python for solving challenging real-world problems, coming up with large scale data collection techniques for training Deep Neural Nets, driving the development of new algorithms that dramatically improve existing methods, researching and maintaining state-of-the-art ML/CV algorithms that can analyze images, and coding full-stack building products from end to end.

$180,000 – $260,000
Undisclosed
YEAR

(USD)

Palo Alto, United States
Maybe global
Onsite
C++
Python
Computer Vision
Deep Learning
NumPy

Head of ML

New
Top rated
Mach9
Full-time
Full-time
Posted

Define and drive a coherent vision for leveraging data to build automation products in surveying and design, translate this vision into a technical roadmap and execute it to advance product capabilities, build and grow the machine learning team including hiring and structuring as the organization scales, mentor ML engineers and researchers by providing technical direction and career growth guidance, stay hands-on by reviewing designs, code, and architecture to maintain credibility and connection with the team, and partner with product and engineering leadership to align research investments with product strategy and customer needs.

Undisclosed

()

San Francisco, United States
Maybe global
Onsite
Python
PyTorch
Machine Learning
Computer Vision
Model Evaluation

ML Engineer

New
Top rated
Mach9
Full-time
Full-time
Posted

Design, train, and evaluate computer vision and 3D ML models for extracting CAD-grade geometry and features from dense LiDAR and imagery. Drive ML research that translates directly into product capabilities by prototyping new approaches, running experiments, and identifying what’s shippable. Own models through the full product lifecycle including problem framing, data strategy, training, evaluation, and final integration into cloud-based CAD software. Develop evaluation methodology and metrics that reflect real surveying and engineering accuracy requirements. Collaborate with ML infrastructure engineers to scale training and inference of models and with product teams to align model behavior with user needs.

Undisclosed

()

San Francisco, United States
Maybe global
Onsite
Python
PyTorch
TensorFlow
Computer Vision
NLP

Data Scientist

New
Top rated
Chattermill
Full-time
Full-time
Posted

The Data Scientist will train, evaluate, and iterate on machine learning models for customer feedback tasks, contributing to the custom fine-tuning pipelines and running experiments with rigorous documentation. They will build and maintain LLM-powered features including retrieval pipelines, reranking systems, and insight generation with guidance from senior team members. They will contribute to evaluation frameworks by helping build test sets, defining metrics, and assessing model quality across classification, extraction, and generative tasks. The role involves working on semantic search and retrieval, developing a strong understanding of embedding-based approaches and beyond, writing clean, well-tested code, and collaborating with Engineering on model integration, data pipelines, and monitoring. Additionally, the Data Scientist will work with the wider Data Science team to translate business and product requirements into practical ML experiments and solutions and stay updated with relevant research to bring useful ideas into team discussions and experiments.

Undisclosed

()

United Kingdom
Maybe global
Remote
Python
PyTorch
Transformers
Model Evaluation
NLP

Want to see more AI Egnineer jobs?

View all jobs

Access all 4,256 remote & onsite AI jobs.

Join our private AI community to unlock full job access, and connect with founders, hiring managers, and top AI professionals.
(Yes, it’s still free—your best contributions are the price of admission.)

Frequently Asked Questions

Need help with something? Here are our most frequently asked questions.

Question text goes here

Lorem ipsum dolor sit amet, consectetur adipiscing elit. Suspendisse varius enim in eros elementum tristique. Duis cursus, mi quis viverra ornare, eros dolor interdum nulla, ut commodo diam libero vitae erat. Aenean faucibus nibh et justo cursus id rutrum lorem imperdiet. Nunc ut sem vitae risus tristique posuere.

[{"question":"What are PyTorch AI jobs?","answer":"PyTorch AI jobs focus on building, training, and deploying deep learning models for applications like computer vision, natural language processing, and generative AI. These positions involve creating custom neural networks, research prototyping with dynamic computation graphs, and transitioning models to production using tools like TorchScript and TorchServe. These roles typically exist in research labs, tech companies, and AI-driven startups."},{"question":"What roles commonly require PyTorch skills?","answer":"Roles that commonly require PyTorch skills include AI researchers, machine learning engineers, data scientists, and deep learning specialists. These professionals develop custom neural networks, implement computer vision solutions, create NLP models, and design predictive analytics systems. They often work on research prototyping and transitioning models to production environments through REST APIs or cloud platforms."},{"question":"What skills are typically required alongside PyTorch?","answer":"Python programming is essential as the framework is deeply integrated with the language. Professionals also need strong foundations in deep learning concepts, familiarity with neural network architectures like CNNs and RNNs, and experience with NumPy. Additional valuable skills include GPU programming with CUDA, distributed training techniques, cloud platforms integration, and knowledge of deployment tools like TorchServe and ONNX Runtime."},{"question":"What experience level do PyTorch AI jobs usually require?","answer":"PyTorch AI jobs span from entry-level to senior positions. Entry roles typically require fundamental Python and deep learning knowledge. Mid-level positions demand practical experience building and deploying models using the framework. Senior roles require extensive experience with complex architectures, distributed training, production deployment, and often specialization in areas like computer vision or NLP."},{"question":"What is the salary range for PyTorch AI jobs?","answer":"Salaries for PyTorch AI jobs vary based on location, experience level, industry, and specific role. Machine learning engineers and AI researchers using this framework typically earn competitive compensation reflecting their specialized skills. Roles involving advanced model development for computer vision, NLP, or generative AI, especially in major tech hubs, command premium compensation packages."},{"question":"Are PyTorch AI jobs in demand?","answer":"PyTorch AI jobs are in high demand across both academia and industry. The framework has gained widespread adoption for cutting-edge research and commercial applications. Many companies seek specialists who can prototype and deploy deep learning models using its dynamic computation graphs. Major cloud providers like Azure, AWS, and Google Cloud have integrated support, further increasing demand for these skills in production environments."},{"question":"What is the difference between PyTorch and TensorFlow in AI roles?","answer":"PyTorch uses dynamic computation graphs allowing for flexible, iterative development and easier debugging, making it popular in research. TensorFlow traditionally used static graphs optimized for production deployment. AI roles focused on research prototyping often prefer PyTorch for its pythonic interface, while production-focused teams might use TensorFlow. However, both frameworks now support both dynamic and static approaches, with the gap narrowing as they evolve."}]