PyTorch Jobs
223 open roles mentioning PyTorch
Backend Engineer - Studio Media Platform
sarvam
Sarvam is building the bedrock of Sovereign AI for India, developing a full-stack AI platform across research, models, infrastructure, and applications. We partner with leading enterprises and public institutions, backed by prominent venture capital firms. We are hiring a Backend Engineer to work on our Studio media platform, which includes AI dubbing, live translation, and foundational services for voice cloning, stem separation, lip sync, and music generation. You will build and maintain production services, ML pipeline libraries, and platform SDKs to enable multilingual media processing at scale for our enterprise customers and Studio users.
Research Engineer, Materials Science
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
Technical Lead Manager, Physical AI
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
Senior AI Infrastructure Engineer - Training Platform
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
Machine Learning Research Engineer, Agent Data Foundation - Enterprise GenAI
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
Machine Learning Systems Research Engineer, Agent Post-training - Enterprise GenAI
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
Tech Lead Manager- MLRE, ML Systems
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
ML Researcher, Foundational Models
sarvam
Sarvam is building the bedrock of Sovereign AI for India, developing a full-stack AI platform focused on making AI genuinely work for India. This role is for a researcher who will tackle open-ended questions about the architecture, optimization, data composition, and training dynamics of our next generation of foundational models. You will have direct access to large compute resources and a tight feedback loop with engineers, driving research from initial hunches to production-ready decisions. This is a hands-on role requiring independent research design, execution at scale, and the ability to translate findings into concrete proposals for production training runs.
Senior Research Engineer, Audio and Speech
decagon
Decagon is seeking a Senior Research Engineer focused on Audio and Speech to build and deploy the next generation of AI voice agents. This role involves advancing multimodal and full-duplex systems that can listen, reason, speak, and respond naturally in real time. You will own your work end-to-end, shipping real improvements and making high-impact technical decisions in a fast-paced, in-office environment.
$200k - $400k
AI engineer
Writer
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.
AI engineer (UK)
Writer
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.
Research Scientist, Foundation Model (Video Generation)
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.
Backend engineer
Nabla
Nabla is seeking a Back-End Engineer to play a critical role in scaling its AI-driven healthcare platform. You will take ownership of key components of our back-end systems, collaborate closely with product and machine learning teams, and help shape the technical direction of our infrastructure. This is an opportunity to contribute end-to-end to a specific part of the product within a lean, fast-moving, and deeply technical engineering environment. Nabla is at the forefront of AI in healthcare, developing solutions that streamline clinical documentation and restore the human connection in patient care, backed by significant recent funding.
Machine Learning Fellow - Human Frontier Collective (UK)
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.
Machine Learning Fellow - Human Frontier Collective (US)
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.
Machine Learning Research Scientist, Reasoning
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
ML Research Engineer, ML Systems
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
Senior / Staff Machine Learning Research Scientist, Agents
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
Machine Learning Fellow - Human Frontier Collective (Canada)
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.
Research Engineer, Multimodal Reasoning For Information Literacy
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.