PyTorch Jobs

Mistral AI

Mistral AI is looking for an Applied AI Engineer to join their customer-facing technical team. This role involves working directly with enterprise clients to deploy cutting-edge AI solutions and address complex technical challenges. You will bridge the gap between AI research and real-world applications, ensuring solutions are robust, scalable, and aligned with customer needs and Mistral's vision. The team operates with a focus on impact and collaboration, valuing direct communication and the best ideas regardless of seniority.

Google DeepMind

Google DeepMind is seeking a Research Engineer to join their materials science team. This role involves accelerating the discovery of new functional materials by integrating artificial intelligence, computational simulation, and automated experimentation. You will collaborate with a diverse interdisciplinary team of domain experts, ML researchers, and engineers. The work focuses on pioneering research in various scientific domains, enabling the validation of early ideas and building infrastructure for promising research lines. You will contribute your scientific domain knowledge to the team's collective expertise.

$141k - $202k

Scale AI

As a Software Engineer on the Machine Learning Infrastructure team, you will build the "Operating System" for our large-scale GPU clusters. You will architect a high-performance training platform that handles the immense complexity of multi-thousand GPU workloads, ensuring every cycle is used efficiently. Your work directly determines the velocity at which our researchers can train and iterate on the world’s most advanced models. The ideal candidate is a systems expert who thrives on solving the orchestration, networking, and reliability challenges that emerge at massive scale. You will partner closely with researchers to build a seamless, resilient environment that transforms raw compute into breakthrough AI.

$216k - $270k

Scale AI

Scale AI is seeking a Technical Lead Manager for its Physical AI team, focusing on the development of general AI that can reason and act in the physical world. This role bridges cutting-edge Machine Learning research with physical robot deployment, leading a team of Research Engineers while remaining a hands-on technical contributor. The primary focus is on developing and evaluating Large-Scale Foundation Models, such as VLAs and World models, to enable robots and autonomous vehicles to generalize across diverse tasks and environments. The team leverages Scale's extensive data infrastructure to help build Foundation Models for Physical AI, aiming to redefine the future of automation.

$249k - $311k

Scale AI

Scale's LLM post-training platform team builds our internal distributed framework for large language model training, powering MLEs, researchers, data scientists, and operators for fast and automatic training and evaluation of LLMs. This platform also serves as the underlying training framework for the data quality evaluation pipeline. You will work closely with Scale’s ML teams and researchers to build the foundation platform which supports all our ML research and development works, optimizing it to enable next generation LLM training, inference, and data curation. If you are excited about shaping the future AI via fundamental innovations, we would love to hear from you!

$265k - $331k

Scale AI

Scale is seeking a Machine Learning Research Engineer to join the Agent Data Foundation team within the Enterprise ML Research Lab. This role will focus on defining and building the data flywheel that drives AI development, particularly for complex agents in enterprise settings. You will conduct cutting-edge research in areas such as synthetic data generation for RL post-training, building agents for production trace analysis, and contributing to a framework for agent construction. The goal is to create state-of-the-art agents through advanced post-training and agent-building algorithms, shaping the future of the GenAI movement.

$265k - $331k

Scale AI

Scale is seeking a Machine Learning Systems Research Engineer to join their Enterprise ML Research Lab. This role will focus on building algorithms for a next-generation Agent RL training platform, supporting large-scale training, and integrating state-of-the-art technologies to optimize ML systems. You will collaborate with other ML researchers and engineers who apply these algorithms to client use cases, including AI cybersecurity firewalls and healthtech search models. If you are passionate about shaping the future of AI, this is an exciting opportunity to contribute to cutting-edge advancements in enterprise GenAI.

$265k - $331k

Abridge

Abridge is seeking an Engineering Manager to lead and grow our Model Inference team. This role is crucial for ensuring our generative AI-powered products are fast, reliable, and world-class. The Inference team is responsible for the end-to-end technical direction of model serving, from architecting low-latency, high-throughput infrastructure to advancing LLM serving techniques. You will lead a team of AI inference engineers, collaborate with ML Research and the AI Platform, and ensure the systems supporting clinician interactions operate at peak efficiency and reliability.

WRITER is seeking an AI Engineer to join their team and shape how enterprises leverage AI. This role involves building tangible AI solutions that power the future of work for leading companies. You will directly impact the performance, scalability, and ethical alignment of cutting-edge LLMs and AI agents, enabling businesses to unlock unprecedented productivity and innovation with AI grounded in their data.

WRITER is a leading enterprise AI platform that empowers businesses to orchestrate AI-powered work and expand human capacity through superintelligence. We build powerful, trustworthy AI solutions that unite IT and business teams for enterprise-wide transformation. Our end-to-end platform enables companies to build and deploy AI agents grounded in their data and fueled by enterprise-grade LLMs. As an AI Engineer, you will be at the forefront of shaping how enterprises harness superintelligence by building tangible AI solutions that power the future of work. Your work will directly impact the performance, scalability, and ethical alignment of our cutting-edge LLMs and AI agents, enabling businesses to unlock unprecedented levels of productivity and innovation.

pika

Pika is pioneering the next generation of creative infrastructure with real-time, multimodal generation and intelligent agentic platforms. We are seeking accomplished Research Scientists in Foundation Models to advance our mission of making agentic, real-time generative technology accessible and transformative for millions of creators. As a key member of our research team, you will design and implement core technologies, develop new methodologies for large-scale multimodal pre-training/mid-training, and drive innovative approaches for foundational model architecture, collaborating closely with engineering and product teams to shape the future of real-time creative and agentic platforms at scale.

Scale AI

The Human Frontier Collective (HFC) Fellowship is a program that brings together top researchers and domain experts to collaborate on high-impact AI projects. As an HFC Fellow, you will apply your expertise to design, evaluate, and interpret advanced generative AI systems, gaining exposure to cutting-edge research and working with a diverse network of leading thinkers. This is a fully remote, independent contractor opportunity with a flexible schedule, offering a chance to contribute to influential ML projects, co-author research publications, and develop your AI expertise within a supportive community.

Scale AI

The Human Frontier Collective (HFC) Fellowship is a fully remote, six-month independent contractor opportunity designed to bring together top researchers and domain experts to collaborate on high-impact AI projects. Fellows will apply their expertise to design, evaluate, and interpret advanced generative AI systems, gaining exposure to cutting-edge research and working with an interdisciplinary network of leading thinkers. This role offers a flexible schedule with 10-40 hour work weeks and competitive pay based on project scope and skillset. Candidates must be authorized to work in the United States, as visa sponsorship is not provided.

Scale AI

This role operates at the forefront of AI research and real-world implementation, with a strong focus on reasoning within large language models (LLMs). You will play a key role in shaping Scale’s data strategy by identifying the most effective data sources and methodologies for improving LLM reasoning. Success in this role requires a deep understanding of LLMs, planning algorithms, and novel approaches to agentic reasoning, as well as creativity in tackling challenges related to data generation, model interaction, and evaluation. You will contribute to impactful research on language model reasoning, collaborate with external researchers, and work closely with engineering teams to bring state-of-the-art advancements into scalable, real-world solutions.

$252k - $315k

Scale AI

Scale's ML platform (RLXF) team builds our internal distributed framework for large language model training and inference. This platform powers MLEs, researchers, data scientists, and operators for fast and automatic training and evaluation of LLMs, as well as data quality evaluation. You will work closely across Scale’s ML teams and researchers to build the foundation platform that supports all our ML research and development, optimizing it to enable the next generation of LLM training, inference, and data curation. If you are excited about shaping the future of AI via fundamental innovations, we would love to hear from you!

$190k - $237k

Scale AI

This role is at the intersection of cutting-edge AI research and practical application, with a focus on studying the data types essential for building state-of-the-art agents, such as browser and SWE agents. The ideal candidate will explore the data landscape needed to advance intelligent, adaptable AI agents, guiding the data strategy at Scale to drive innovation. This position requires not only expertise in LLM agents and planning algorithms but also creativity in addressing novel challenges related to data, interaction, and evaluation. You will contribute to impactful research publications on agents, collaborate with customer researchers, and work alongside the engineering team to translate these advancements into real-world, scalable solutions.

$302k - $378k

Scale AI

The Human Frontier Collective (HFC) Fellowship is a six-month, fully remote, independent contractor opportunity designed to bring together top researchers and domain experts to collaborate on high-impact AI projects. Fellows will apply their expertise to design, evaluate, and interpret advanced generative AI systems, gaining exposure to cutting-edge research and working with an interdisciplinary network of leading thinkers. This role offers a flexible schedule, allowing fellows to set their own hours (10-40 hours per week), and provides opportunities for professional development through review projects, advisory roles, and research, contributing to academic visibility and professional recognition.

Google DeepMind

Google DeepMind's research team is dedicated to tackling complex challenges in online information quality, advancing the state of the art by developing innovative solutions to detect manipulated media and misleading narratives. The team leverages interdisciplinary work spanning provenance analysis and the creation of tools for AI-assisted information literacy, with a focus on ensuring the integrity of digital discourse and a safer online environment. This role involves researching and building multimodal reasoning systems and Vision-Language Models (VLMs) to assess the trustworthiness of media on the internet, with a passion for advancing information literacy using machine learning and computational techniques.

Mistral AI

Mistral AI is seeking a Technical Lead, Applied AI to drive the technical strategy, execution, and delivery of complex AI solutions for enterprise customers. This role involves leading project teams of Applied AI Engineers to ensure successful deployment of Mistral AI products and the development of high-impact, scalable AI use cases. You will serve as the primary technical point of contact for strategic customers, guiding them through the entire lifecycle from pre-sales to post-implementation, while collaborating with research, product, and engineering teams to shape future offerings. The position bridges cutting-edge AI research with real-world enterprise applications, ensuring solutions are robust, scalable, and aligned with customer needs and Mistral's technological vision.

thinkingmachines

Thinking Machines Lab is seeking pre-training researchers to join their mission of advancing collaborative general intelligence. This role is central to developing the next generation of AI models by blending research with large-scale data engineering. You will be responsible for assembling pre-training datasets and data systems, designing and implementing methods for sourcing, curating, and analyzing data for quality and performance. The position involves working with automated pipelines and human-in-the-loop processes, contributing both scientific insights and production-grade code. It's an ideal opportunity for individuals passionate about the intersection of data, machine learning, and systems, and who are eager to shape the future of AI.

$350k - $475k