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

Software Engineer

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

Develop low-level drivers for the sensors or actuators of robots, develop the OS and middleware of the robots, integrate embedded algorithms such as Guidance, Navigation, Control, and Computer Vision, optimize runtime of algorithms on various hardware accelerators like GPU, TPU, DSp, develop the backbone of a command and control system for massive data ingestion and processing, develop a web-based front-end to display theatre of operations and allow mission conduction, build internal tools to improve efficiency and reduce technical debt, develop connectors between existing company tools like ERP, MES, PLM, implement code into production-ready environments, ensure seamless integration with Harmattan AI’s systems, conduct rigorous code reviews, test algorithms in real-world environments, develop monitoring tools, track model performance and continuously improve deployed solutions, collaborate closely with other hardware and software teams to align development with system requirements, and communicate findings effectively to stakeholders.

Undisclosed

()

Morocco
Maybe global
Onsite
Python
C++
PyTorch
Computer Vision
CUDA

Senior Staff Research Scientist, Speech Technologies

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

Design, develop, and iterate on data-driven ASR models for streaming and non-streaming conversational speech applications; research and implement state-of-the-art end-to-end speech recognition architectures tailored to the medical domain; train, evaluate, and optimize ASR models across accuracy, latency, and resource utilization dimensions; preprocess and curate large-scale speech datasets to support robust model training; collaborate closely with LLM, product, and clinical teams to integrate speech technologies into the broader Hippocratic AI platform; contribute to the team's research culture through experimentation, documentation, and knowledge sharing.

Undisclosed

()

Bellevue or Menlo Park, United States
Maybe global
Hybrid
Python
C++
PyTorch
TensorFlow
NLP

ML/AI Engineer - Vehicle Intelligence

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

Develop AI-powered vehicle intelligence features that understand user intent, trip goals, vehicle state, and system constraints. Apply reinforcement learning, planning, optimization, and data-driven modeling to improve vehicle-level decisions across energy, comfort, charging, routing, and proactive vehicle preparation. Build models using vehicle telemetry, navigation data, user behavior, weather, traffic, cabin conditions, charging patterns, and fleet data. Create personalization models that learn user routines, comfort preferences, driving patterns, charging habits, and trip priorities while preserving privacy and user control. Use simulation, digital twins, and scenario-based testing to train, evaluate, and validate AI behavior before production deployment. Collaborate with autonomous driving and VLA teams to define interfaces for sharing user intent, route objectives, vehicle constraints, energy targets, comfort preferences, and system-level recommendations. Integrate ML models into production vehicle and cloud platforms, considering latency, compute efficiency, reliability, safety, explainability, and over-the-air update readiness. Work cross-functionally with Product, UX, Systems Engineering and Controls.

$220,780 – $311,220
Undisclosed
YEAR

(USD)

Sunnyvale or San Francisco, United States
Maybe global
Onsite
Python
PyTorch
TensorFlow
JAX
Reinforcement Learning

Member of Technical Staff (Machine Learning Engineer)

New
Top rated
Reka
Full-time
Full-time
Posted

Translate cutting-edge research into production-ready machine learning systems. Design, build, and deploy end-to-end ML models and pipelines. Develop and optimize models for image and video processing. Own the full ML lifecycle including experimentation, training/fine-tuning, evaluation, and deployment. Rapidly prototype using open-source models and adapt them for product needs. Conduct experiments, analyze results, and iterate to improve performance. Collaborate with researchers and cross-functional teams (product, engineering, design) to deliver ML solutions at scale. Participate with advancements in machine learning and apply them to continuously improve products.

Undisclosed

()

Maybe global
Remote
Python
Java
C++
PyTorch
TensorFlow

AI Research Resident

New
Top rated
Maincode
Full-time
Full-time
Posted

Lead research that advances Maincode's work on capable, useful, and trustworthy AI systems. Design and execute experiments, develop new research directions, and collaborate closely with researchers and engineers. Produce research outputs suitable for top-tier conferences, journals, technical reports, open-source releases, or deployment in Matilda and future Maincode systems.

Undisclosed

()

Australia
Maybe global
Remote
Python
PyTorch
TensorFlow
Reinforcement Learning
Model Evaluation

Manager, Deployment Engineering

New
Top rated
Armada
Full-time
Full-time
Posted

The responsibilities include translating business requirements into requirements for AI/ML models, preparing data to train and evaluate AI/ML/DL models, building AI/ML/DL models using state-of-the-art algorithms especially transformers, testing and evaluating the AI/ML/DL models, publishing the models, datasets, and evaluations, deploying models in production by containerizing them, working with customers and internal employees to refine model quality, establishing continuous learning pipelines for models with online or transfer learning, and building and deploying containerized applications on cloud or on-premise environments.

$154,560 – $193,200
Undisclosed
YEAR

(USD)

Bellevue
Maybe global
Remote
Python
Java
C++
PyTorch
TensorFlow

Researcher, Agent Post-Training, Personality

New
Top rated
OpenAI
Full-time
Full-time
Posted

As a member of the Agent Post-training Personality team, the role involves helping to make OpenAI’s agents exceptional collaborators by studying what makes an agent thoughtful, clear, perceptive, appropriately proactive, and easy to work with. This includes translating those insights into evaluations, training data, reward signals, and model improvements. Responsibilities include developing a rigorous understanding of effective agent collaboration across various types of work, turning qualitative judgments about model behavior into concrete hypotheses, evaluations, graders, and training interventions, studying user signals to understand behaviors that create trust and satisfaction, working with human experts and trainers to produce high-quality data capturing excellent collaborative behavior, improving reward models and reinforcement learning objectives, collaborating with pretraining and early-training teams on data and objectives, building pipelines for updating training data, partnering with product teams to turn consumer insights into model improvements, and owning projects end to end from identifying behavioral failures through experimentation, training, evaluation, and launch.

$295,000 – $445,000
Undisclosed
YEAR

(USD)

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

Deployment Engineer

New
Top rated
Armada
Full-time
Full-time
Posted

Translate business requirements into AI/ML model requirements. Prepare data to train and evaluate AI/ML/DL models. Build AI/ML/DL models using state-of-the-art algorithms, especially transformers, sometimes leveraging existing algorithms from research. Test and evaluate models, benchmark quality, and publish models, datasets, and evaluations. Deploy models in production by containerizing them. Work with customers and internal employees to refine model quality. Establish continuous learning pipelines for models with online 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

Research Engineers, Post-Training

New
Top rated
Distyl
Full-time
Full-time
Posted

Research Engineers design and run post-training workflows that improve the behavior, reliability, and usefulness of AI systems. They develop datasets, preference signals, evaluation suites, reward models, fine-tuning workflows, and feedback loops for applied AI use cases. They investigate how different post-training techniques affect system behavior across enterprise workflows and production constraints. They build infrastructure for experimentation, model comparison, regression testing, and behavior analysis. Research Engineers partner with AI Researchers to explore new post-training methods and with AI Engineers to apply successful techniques in deployed systems. They analyze model outputs, failure modes, human feedback, and production traces to identify opportunities for behavioral improvement. They create repeatable processes for adapting AI systems to customer domains while preserving robustness, transparency, and maintainability. They communicate clearly with internal teams and customer stakeholders about model behavior, evaluation results, limitations, and tradeoffs.

$150,000 – $250,000
Undisclosed
YEAR

(USD)

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

Deployment Lead

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. You will develop data pipelines from diverse sources such as code repositories, web browsers, and computer systems. You will implement and adapt popular open-source agent libraries and benchmarks with proprietary datasets and models. You will engage 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. You will collaborate closely with frontier AI lab customers to understand their requirements and guide model development. Additionally, you will 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
Prompt Engineering

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