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 The Homebase

Technical Program Manager, R&D & Technology Transfer

New
Top rated
Intrinsic
Full-time
Full-time
Posted

Lead the research and development of novel deep learning algorithms that enable robots to perform complex, contact-rich manipulation tasks. Explore the intersection of computer vision and robotic control by designing systems that allow robots to perceive and interact with objects in dynamic environments. Create models that integrate visual data to guide physical manipulation, moving beyond simple grasping to sophisticated handling of diverse items. Collaborate with a multidisciplinary team of engineers and researchers to translate cutting-edge concepts into robust capabilities deployable on physical hardware for industrial applications. Research and develop deep learning architectures for visual perception and sensorimotor control in contact-rich scenarios. Design algorithms for high precision manipulation of complex or deformable objects. Work with software engineers to optimize and deploy research prototypes onto physical robotic hardware. Evaluate model performance in simulation and real-world environments to ensure robustness and reliability. Identify opportunities to apply advancements in computer vision and robot learning to practical industrial problems. Mentor junior researchers and contribute to the technical direction of the manipulation research roadmap.

Undisclosed

()

Mountain View, United States
Maybe global
Onsite
Python
C++
PyTorch
TensorFlow
JAX

2026 New Grad | Software Engineer, Full-Stack

New
Top rated
Loop
Full-time
Full-time
Posted

Ship critical infrastructure managing real-world logistics and financial data for large enterprises. Own the why by building deep context through customer calls and understanding Loop's value to customers, pushing back on requirements if better solutions exist. Work full-stack across system boundaries including frontend UX, LLM agents, database schema, and event infrastructures. Leverage AI tools to handle routine tasks enabling focus on quality, architecture, and product taste. Constantly optimize development loops, refactor legacy patterns, automate workflows, and fix broken processes to raise velocity.

$150,000 – $150,000
Undisclosed
YEAR

(USD)

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

Member of Technical Staff - Post Training, Applied

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

The role involves acting as the technical owner for enterprise customer post-training engagements, owning post-training projects end-to-end from customer requirements through delivery and evaluation. Responsibilities include translating customer requirements into concrete post-training specifications and workflows, designing and executing data generation, filtering, and quality assessment processes, running supervised fine-tuning, preference alignment, and reinforcement learning workflows, as well as designing task-specific evaluations, interpreting results, and feeding learnings back into core post-training pipelines.

Undisclosed

()

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

Senior Machine Learning Engineer - Payments

New
Top rated
Plaid
Full-time
Full-time
Posted

As a machine learning engineer on the core ML payments team, you will design, build, and deploy scalable machine learning solutions and systems. You will experiment with new modeling approaches and strategies, collaborate closely with a team of engineers on ingesting signals, and productionize these models. Your work will empower millions of users through well-known and emerging Fintech applications with access to financial services. Responsibilities also include working on both 0-1 stage problems and 1-10 stage problems, developing AI/ML models through the full lifecycle from offline training to online serving and monitoring, collaborating with teams across the company to define the ML roadmap, and applying data-driven decisions in day-to-day work in a high ownership, bottom-up driven team.

$225,600 – $337,200
Undisclosed
YEAR

(USD)

San Francisco, United States
Maybe global
Remote
Python
PyTorch
TensorFlow
Scikit-learn
NumPy

Machine Learning Engineer - Perception Mapping (copy)

New
Top rated
Zoox
Full-time
Full-time
Posted

As a software engineer on the perception mapping team at Zoox, you will curate, validate, and label datasets for model training and validation. You will research, implement, and train machine learning models to perform semantic map element detection and closely collaborate with validation teams to formulate and execute model validation pipelines. You will integrate models into the greater onboard autonomy system within compute budgets. Additionally, you will serve as a technical leader on the team, maintaining coding and ML development best practices and contributing to architectural decisions.

$189,000 – $227,000
Undisclosed
YEAR

(USD)

Foster City, United States
Maybe global
Onsite
Python
PyTorch
NumPy
C++
Computer Vision

Research Engineer/Scientist - Generative UI, Consumer Products

New
Top rated
OpenAI
Full-time
Full-time
Posted

As a Research Engineer/Scientist on the Consumer Products Research team, you will train and evaluate state-of-the-art models along important axes for future devices, work through challenges to transform nascent research capabilities into usable capabilities, and help define software frameworks for long-term future use.

$380,000 – $460,000
Undisclosed
YEAR

(USD)

San Francisco, United States
Maybe global
Hybrid
Python
PyTorch
TensorFlow
NLP
Prompt Engineering

Senior Research Engineer/Scientist - Edge, Consumer Products

New
Top rated
OpenAI
Full-time
Full-time
Posted

As a Research Engineer/Scientist on the Consumer Products Research team, you will train and evaluate multimodal state-of-the-art (SoTA) models along axes important to the vision for future devices. You will develop novel architectures that improve model performance when scaling the models themselves is not an option. You will also work to transition nascent research capabilities into capabilities that can be built upon.

$380,000 – $460,000
Undisclosed
YEAR

(USD)

San Francisco, United States
Maybe global
Hybrid
Python
PyTorch
TensorFlow
Model Evaluation
Transformers

Software Engineer

New
Top rated
AIFund
Full-time
Full-time
Posted

Design, develop, and maintain web applications and backend services that integrate ML-powered features. Collaborate closely with Machine Learning Engineers and Product Managers to understand ML system requirements and translate them into robust software solutions. Build reliable, scalable, and low-latency services that support ML inference, data workflows, and AI-driven user experiences. Use LLMs to build scalable and reliable AI agents. Own the full software development lifecycle: design, implementation, testing, deployment, monitoring, and maintenance. Ensure high standards for code quality, testing, observability, and operational excellence. Troubleshoot production issues and participate in on-call or support rotations when needed. Mentor junior engineers and contribute to technical best practices across teams. Act as a strong cross-functional partner between product, engineering, and ML teams.

Undisclosed

()

San Francisco Bay Area, United States
Maybe global
Hybrid
Python
Docker
Kubernetes
AWS
GCP

Applied AI, Evaluation Engineer

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

Design and implement comprehensive evaluation frameworks to measure LLM capabilities across diverse customer use cases including text generation, reasoning, code, and domain-specific applications; build scalable evaluation infrastructure and pipelines that enable rapid, reproducible assessment of model performance; develop novel evaluation methodologies to assess emerging capabilities or verticalized use cases such as cybersecurity, finance, and healthcare; create custom evaluation suites tailored to enterprise customers' specific needs while working closely with them to understand their requirements and success criteria; collaborate with research teams to translate evaluation insights into model improvements and training decisions; partner with product teams to continuously improve evaluation tooling based on customer feedback.

Undisclosed

()

Paris, France
Maybe global
Onsite
Python
PyTorch
Hugging Face
Transformers
Prompt Engineering

Applied AI, AI Engineer for Mistral

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

The Applied AI engineer at Mistral works within the customer-facing technical team to deploy AI solutions that deliver measurable business impact. Responsibilities include identifying high-value internal use cases across various departments such as engineering, legal, HR, sales, and operations; building end-to-end LLM applications including prompts, RAG pipelines, APIs, simple UIs, deployment, and monitoring; owning the full lifecycle of AI tools from prototype to production, maintenance, and iteration; documenting learnings and sharing insights with product and research teams; and converting successful internal tools into customer demos or case studies where appropriate. The role also involves acting as the first internal customer for these tools to identify edge cases and limitations, improving models through usage feedback.

Undisclosed

()

Paris, France
Maybe global
Onsite
Python
React
PyTorch
Hugging Face
Transformers

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