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

Machine Learning Engineer

New
Top rated
Noetica
Full-time
Full-time
Posted

As a Machine Learning Engineer at Noetica, you will build ML models and pipelines with scalability and reproducibility as foundational principles, develop NLP systems that can accurately process and understand complex legal language and terminology, and design and implement LLM-based solutions that are well-documented and empower legal professionals to extract valuable insights. You will extend and create reliable model evaluation frameworks to ensure accuracy and reduce model drift or bias, simplify complex ML systems into more manageable solutions, optimize model performance through smart feature engineering and efficient algorithm selection based on actual use cases, and work with security engineers to implement responsible AI practices that protect sensitive data while delivering valuable insights.

$187,000 – $270,000
Undisclosed
YEAR

(USD)

New York, United States
Maybe global
Hybrid
Python
PyTorch
TensorFlow
NLP
Model Evaluation

MEP Manager, Data Centers

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

Develop novel architectures, system optimizations, optimization algorithms, and data-centric optimizations that significantly improve over state-of-the-art. Take advantage of the computational infrastructure of Together to create the best open models in their class. Understand and improve the full lifecycle of building open models; release and publish insights through blogs, academic papers, etc. Collaborate with cross-functional teams to deploy models and make them available to a wider community and customer base. Stay up-to-date with the latest advancements in machine learning.

$160,000 – $230,000
Undisclosed
YEAR

(USD)

San Francisco
Maybe global
Onsite
Python
PyTorch
TensorFlow
Machine Learning
Model Evaluation

Solutions Engineer (AI/ML, Pre-Sales)

New
Top rated
DatologyAI
Full-time
Full-time
Posted

The Solutions Engineer (AI/ML, Pre-Sales) will work closely with strategic customers to understand their data curation needs, business challenges, and technical requirements. The role involves leading end-to-end customer proofs of concept (PoCs) that connect data curation to training behavior and evaluation outcomes, including dataset analysis, training plan design, and interpreting results. They will partner with customer machine learning teams to map data and curation strategies, design and execute evaluation plans for base and post-trained models, select appropriate benchmarks and metrics, and run model evaluations. Additionally, the engineer will produce customer-ready evaluation reports detailing methodology, metrics, baselines, ablations (e.g., curated vs raw data), conclusions, and recommendations for productionization. They must communicate technical results effectively to both ML experts and executive stakeholders, explaining tradeoffs in compute, latency, and deployment cost. Collaboration with go-to-market, engineering, and research teams is essential to deliver compelling demos, align on requirements, and incorporate customer insights into model training and product strategies. The role also includes providing technical guidance, training, and documentation to enable prospects to confidently assess the solution.

$230,000 – $300,000
Undisclosed
YEAR

(USD)

Redwood City, United States
Maybe global
Onsite
Python
PyTorch
Hugging Face
Distributed Training
Cloud Platforms

Machine Learning Engineer, Applied AI

New
Top rated
Ideogram
Full-time
Full-time
Posted

The Machine Learning Engineer is responsible for leading applied AI initiatives by bridging research and product to turn generative models into production features across the first-party app and API. Responsibilities include experimenting rapidly, building rigorous evaluations and datasets, partnering with research, engineering, infrastructure, and product teams to ship reliable and scalable ML systems. They will fine-tune and deploy models for creative use cases such as text-to-image, image-to-text, image enhancement and editing, and multimodal applications. The engineer sets clear success metrics including quality, latency, and cost, and contributes to the safety, monitoring, and reliability of the systems. They lead projects from 0 to 1 that shape Applied AI practices at Ideogram while delivering features that bring value and delight to users.

Undisclosed

()

New York, United States
Maybe global
Remote
Python
PyTorch
JAX
Transformers
Model Evaluation

Product Security Applied AI Intern, Summer 2026

New
Top rated
Crusoe
Intern
Full-time
Posted

Assist in designing and implementing custom large language models (LLMs) and fine-tuning models for specific tasks. Build and experiment with agent libraries and workflow orchestration frameworks. Explore neo-cloud technologies, containerized environments, and virtualized infrastructure. Learn and apply security and privacy best practices in AI pipelines and deployments. Collaborate with the team to document, test, and optimize agent behaviors and models. Participate in knowledge sharing and mentorship sessions to gain exposure to AI, cloud, and security tradecraft.

$1,905 – $1,905 / week
Undisclosed
WEEK

(USD)

San Francisco, United States
Maybe global
Onsite
Python
PyTorch
TensorFlow
OpenAI API
Hugging Face

Mechanical Engineer - Hands

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

Design, deploy, and maintain Figure's training clusters. Architect and maintain scalable deep learning frameworks for training on massive robot datasets. Work together with AI researchers to implement training of new model architectures at a large scale. Implement distributed training and parallelization strategies to reduce model development cycles. Implement tooling for data processing, model experimentation, and continuous integration.

$150,000 – $350,000
Undisclosed
YEAR

(USD)

San Jose, United States
Maybe global
Onsite
Python
PyTorch
AWS
GCP
Kubernetes

Lead Machine Learning Engineer

New
Top rated
Fyxer
Full-time
Full-time
Posted

The Lead Machine Learning Engineer will own the development and improvement of the system predicting the next action salespeople should take to advance their relationships. Responsibilities include selecting the best model architecture and approach, involving a mixture of LLM steps and traditional ML models, picking evaluation metrics, designing systems to analyze models in production to identify areas for improvement, and identifying when to use the human data team for training or validation datasets. The engineer will read relevant research to find the best approach for their use case and, in partnership with the CTO, define how machine learning works with product engineering, model operations, and human data teams and how the team should develop moving forward.

£200,000 – £200,000
Undisclosed
YEAR

(GBP)

London, United Kingdom
Maybe global
Hybrid
Python
PyTorch
TensorFlow
Hugging Face
Transformers

Lead Machine Learning Engineer

New
Top rated
Faculty
Full-time
Full-time
Posted

Set the technical direction for complex machine learning projects, balancing trade-offs and guiding team priorities. Design, implement, and maintain reliable, scalable ML and software systems while justifying key architectural decisions. Define project problems, develop roadmaps, and oversee delivery across multiple workstreams in often ill-defined, high-risk environments. Drive the development of shared resources and libraries across the organisation and guide other engineers in contributing to them. Lead hiring processes, make informed selection decisions, and mentor multiple individuals to foster team growth. Proactively develop and execute recommendations for adopting new technologies and changing ways of working to stay competitive. Act as a technical expert and coach for customers, accurately estimate large workstreams, and defend rationale to stakeholders.

Undisclosed

()

London, United Kingdom
Maybe global
Hybrid
Python
Scikit-learn
TensorFlow
PyTorch
Docker

Software Engineer, macOS Core Product - Palm Coast, USA

New
Top rated
Speechify
Full-time
Full-time
Posted

Work alongside machine learning researchers, engineers, and product managers to bring AI Voices to customers for diverse use cases. Deploy and operate the core ML inference workloads for the AI Voices serving pipeline. Introduce new techniques, tools, and architecture that improve performance, latency, throughput, and efficiency of deployed models. Build tools to provide visibility into bottlenecks and sources of instability, and design and implement solutions to address the highest priority issues.

$140,000 – $200,000
Undisclosed
YEAR

(USD)

Palm Coast, United States
Maybe global
Remote
Python
GCP
Docker
Kubernetes
MLflow

Software Engineer, macOS Core Product - Seattle, USA

New
Top rated
Speechify
Full-time
Full-time
Posted

Work alongside machine learning researchers, engineers, and product managers to bring AI Voices to customers for various use cases. Deploy and operate the core machine learning inference workloads for the AI Voices serving pipeline. Introduce new techniques, tools, and architecture to improve the performance, latency, throughput, and efficiency of deployed models. Build tools to identify bottlenecks and sources of instability and design and implement solutions to address the highest priority issues.

$140,000 – $200,000
Undisclosed
YEAR

(USD)

Seattle, United States
Maybe global
Remote
Python
GCP
Docker
Kubernetes
MLflow

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