TensorFlow Jobs

Cohere

Cohere is a leading enterprise AI company building cutting-edge foundation AI models and end-to-end products. The GPU Clusters team is central to Cohere's infrastructure, responsible for building and operating the superclusters that power our frontier AI models. This role involves enabling research and development at the intersection of cutting-edge hardware, distributed systems, and AI research. As an Engineering Manager, you will lead a team of highly motivated engineers passionate about GPU infrastructure and AI, fostering a culture of technical excellence and innovation in a remote-first environment. This is a unique opportunity to shape the infrastructure powering the next generation of AI.

Anthropic

As a Research Engineer within Reinforcement Learning, you will collaborate with a diverse group of researchers and engineers to advance the capabilities and safety of large language models. This role blends research and engineering responsibilities, requiring you to both implement novel approaches and contribute to the research direction. You'll work on fundamental research in reinforcement learning, creating 'agentic' models via tool use for open-ended tasks such as computer use and autonomous software generation, improving reasoning abilities in areas such as mathematics, and developing prototypes for internal use, productivity, and evaluation.

Anthropic

As a Research Engineer within Reinforcement Learning, you will collaborate with a diverse group of researchers and engineers to advance the capabilities and safety of large language models. This role blends research and engineering responsibilities, requiring you to both implement novel approaches and contribute to the research direction. You'll work on fundamental research in reinforcement learning, creating 'agentic' models via tool use for open-ended tasks such as computer use and autonomous software generation, improving reasoning abilities in areas such as mathematics, and developing prototypes for internal use, productivity, and evaluation.

pika

Pika is shaping the future of creative infrastructure with real-time, multimodal generation and intelligent agentic platforms. We are seeking passionate Research Interns at the BS, MS, or PhD level with significant experience working on Videogen to join our team and make an immediate impact. In this role, you will directly contribute to mid-training and post-training efforts for Videogen models and support advanced generative systems. This internship is ideal for students and researchers who have hands-on Videogen experience, are excited to work at the intersection of generative AI and multimedia, and are eager to make meaningful contributions alongside top AI talent.

pika

We are seeking a Staff/Lead Software Engineer, AI Infrastructure, to play a critical role in building and scaling the core infrastructure that powers Pika’s AI capabilities. In this position, you will lead the design and implementation of GPU infrastructure, AI model serving APIs, and general AI infrastructure execution, enabling cutting-edge machine learning features that drive our products. You will be responsible for architecting robust, distributed systems optimized for high-performance AI workloads, large-scale GPU orchestration, and low-latency, reliable API serving. Your work will directly impact the way users experience and interact with generative AI at scale. As a senior technical leader, you’ll also mentor engineers, drive best practices, and set the technical vision for AI infrastructure at Pika.

Scale AI

We are seeking a Machine Learning Platform Engineer to develop internal tooling for fine-tuning and evaluating large language models. This role is crucial for advancing our capabilities in working with cutting-edge AI technologies.

$140k - $190k

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

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 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.

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

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

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.

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

thinkingmachines

Thinking Machines Lab is seeking researchers to bridge the gap between raw AI intelligence and useful, safe, and collaborative systems. This role focuses on post-training data research, combining human insight and machine learning techniques to capture and steer model behavior based on human preferences. You will be responsible for translating research ideas into actionable data through labeling and collection campaigns, understanding data quality science, and developing metrics to measure the impact of data and training interventions. The position also involves exploring new paradigms for human-AI interaction and scalable oversight, blending research, data operations, and technical implementation to advance human-centered AI systems. This role requires both fundamental research and practical engineering, making it ideal for individuals who enjoy deep theoretical exploration and hands-on experimentation.

$350k - $475k

Cohere

Cohere is seeking a Research Intern to collaborate with researchers and tools on designing and implementing novel research ideas and shipping state-of-the-art models to production. Interns will have the opportunity to work on various teams covering base model training, retrieval augmented generation, data and evaluation, safety, and finetuning, or any research area relating to LLMs. This role offers a chance to broaden research connections while gaining deep experience in a growing AI startup.