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

Electrical Design Engineer

New
Top rated
Armada
Full-time
Full-time
Posted

Translate business requirements into requirements for AI/ML models; prepare data to train and evaluate AI/ML/DL models; build AI/ML/DL models by applying state-of-the-art algorithms, especially transformers; leverage existing algorithms from academic or industrial research when applicable; test, evaluate, and benchmark the AI/ML/DL models, and publish the models, data sets, and evaluations; deploy models in production by containerizing the models; work with customers and internal employees to refine the quality of the models; establish continuous learning pipelines for models with online learning or transfer learning; build and deploy containerized applications on cloud or on-premise environments.

$154,560 – $193,200
Undisclosed
YEAR

(USD)

United States
Maybe global
Remote
Python
Java
C++
PyTorch
TensorFlow

Machine Learning Engineer

New
Top rated
HappyRobot
Full-time
Full-time
Posted

Design, build, and maintain scalable machine learning systems including data ingestion, preprocessing, training, testing, and deployment. Develop and optimize end-to-end ML pipelines encompassing data collection, labeling, training, validation, and monitoring to ensure reliability and reproducibility. Implement robust MLOps practices such as model versioning, experiment tracking, CI/CD for machine learning, and continuous monitoring in production environments. Collaborate with product and engineering teams to integrate and deploy models into real-time products with a focus on efficiency and scalability. Ensure data quality, observability, and performance across all AI systems. Stay current with the latest AI infrastructure, tooling, and research to support ongoing innovation.

Undisclosed

()

Spain
Maybe global
Remote
Python
Go
MLOps
MLflow
Docker

Software Engineer (SF)

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

Work on a small, high-caliber team building AI products for clients, from requirements gathering and prototyping through system design, development, testing, and deployment. Own features end-to-end and develop domain expertise across a range of AI use cases. Spend most of the time coding and frequently interact with clients to ensure the solutions meet their needs.

$160,000 – $220,000
Undisclosed
YEAR

(USD)

San Francisco, United States
Maybe global
Hybrid
Python
JavaScript
TypeScript
PyTorch
TensorFlow

Senior / Staff Software Engineer (SF/NY)

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

You will work on a small, high-caliber team building AI products for clients, setting technical direction, writing code, and serving as the go-to person when challenges arise. Spend approximately 75% of your time coding and 25% interacting with clients, including CTOs, to understand problems, evaluate tradeoffs, and ensure solutions meet their needs.

$230,000 – $350,000
Undisclosed
YEAR

(USD)

San Francisco or New York, United States
Maybe global
Hybrid
Python
JavaScript
TypeScript
PyTorch
TensorFlow

Researcher, Misalignment Research

New
Top rated
OpenAI
Full-time
Full-time
Posted

Design and implement worst-case demonstrations that concretely reveal AGI alignment risks for stakeholders, focusing on high-stakes use cases; develop adversarial and system-level evaluations based on these demonstrations and promote their adoption across OpenAI; create automated tools and infrastructure to scale automated red-teaming and stress testing; conduct research on failure modes of alignment techniques and propose improvements; publish influential internal or external papers that impact safety strategy or industry practice; collaborate with engineering, research, policy, and legal teams to integrate findings into product safeguards and governance; and mentor engineers and researchers to foster a culture of rigorous, impact-oriented safety work.

$295,000 – $445,000
Undisclosed
YEAR

(USD)

San Francisco, United States
Maybe global
Onsite
Python
PyTorch
TensorFlow
Model Evaluation
Reinforcement Learning

Researcher, Alignment Science

New
Top rated
OpenAI
Full-time
Full-time
Posted

As a Research Engineer / Research Scientist on the Alignment team, you will design and implement alignment experiments focused on intent following, honesty, calibration, and robustness. You will train and evaluate models using reinforcement learning and other empirical machine learning methods. Your role includes developing evaluations for failure modes such as hallucination, instruction-following failures, reward hacking, covert actions, and scheming. You will study methods that encourage models to verify their behavior and report shortcomings honestly, including confession-style training objectives. You will build monitoring and inference-time interventions that ensure compliant behavior or surface model issues to users or downstream systems. Additionally, you will investigate how alignment methods scale with model capability, compute, data, context length, action length, and adversarial pressure. You will integrate successful techniques into model training and deployment workflows, produce externally publishable research when results advance the broader science of alignment, and collaborate with researchers and engineers across post-training, reinforcement learning, evaluations, safety, and product-facing teams.

$250,000 – $445,000
Undisclosed
YEAR

(USD)

San Francisco, United States
Maybe global
Hybrid
Python
PyTorch
Reinforcement Learning
Model Evaluation
NLP

AI/ML Engineer

New
Top rated
Air Apps
Full-time
Full-time
Posted

Develop, train, and optimize machine learning models for various mobile app features. Research and implement state-of-the-art AI techniques to improve user engagement and app performance. Collaborate with cross-functional teams to integrate AI-driven solutions into applications. Design and maintain scalable ML pipelines, ensuring efficient model deployment and monitoring. Analyze large datasets to derive insights and drive data-driven decision-making. Stay updated with the latest AI trends and best practices and incorporate them into development processes. Optimize AI models for mobile environments to ensure high performance and low latency.

€60,000 – €76,000
Undisclosed
YEAR

(EUR)

Barcelona, Spain
Maybe global
Remote
Python
TensorFlow
PyTorch
NLP
Computer Vision

Machine Learning Enginer, Core Evaluations

New
Top rated
Cantina Labs
Full-time
Full-time
Posted

The responsibilities include designing model evaluation pipelines for models in both development and production environments, designing user studies for subjective model evaluations, converting requirements into measurable metrics, and designing and developing automated evaluation dashboards to monitor and compare model performance. It also involves training new models to capture various evaluation metrics, communicating with the model team to help design improved models based on evaluation results, coordinating with the data team to determine necessary data for enhancing model performance, collaborating with the product manager to ensure product requirements are accurately measured, helping to grow the evaluation team as the founding member, and leading the evaluation team in the future.

Undisclosed

()

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

AI/ML Engineer

New
Top rated
Air Apps
Full-time
Full-time
Posted

Develop, train, and optimize machine learning models for various mobile app features. Research and implement state-of-the-art AI techniques to improve user engagement and app performance. Collaborate with cross-functional teams to integrate AI-driven solutions into applications. Design and maintain scalable ML pipelines, ensuring efficient model deployment and monitoring. Analyze large datasets to derive insights and drive data-driven decision-making. Stay updated with the latest AI trends and best practices, incorporating them into development processes. Optimize AI models for mobile environments to ensure high performance and low latency.

€60,000 – €76,000
Undisclosed
YEAR

(EUR)

Helsinki, Finland
Maybe global
Remote
Python
TensorFlow
PyTorch
NLP
Computer Vision

AI/ML Engineer

New
Top rated
Air Apps
Full-time
Full-time
Posted

Develop, train, and optimize machine learning models for various mobile app features. Research and implement state-of-the-art AI techniques to improve user engagement and app performance. Collaborate with cross-functional teams to integrate AI-driven solutions into applications. Design and maintain scalable ML pipelines, ensuring efficient model deployment and monitoring. Analyze large datasets to derive insights and drive data-driven decision-making. Stay updated with the latest AI trends and best practices, incorporating them into development processes. Optimize AI models for mobile environments to ensure high performance and low latency.

€60,000 – €76,000
Undisclosed
YEAR

(EUR)

Munich, Germany
Maybe global
Remote
Python
TensorFlow
PyTorch
NLP
Computer Vision

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."}]