TensorFlow Jobs

Anthropic

The Safeguards ML Inference Path team designs, builds, and operates the production infrastructure that powers Claude's ML based safety systems. We collaborate closely with safety researchers and inference engineers to bring new classifiers and novel classes of ML defenses to production. We own the research → production transfer of new safety technologies that is on the critical path for every Claude model launch. We build for scale, serving thousands of ML classifiers for all requests on the token generation path and for every platform Claude runs on. We are looking for engineers who have deep expertise in productionizing ML systems, working at the intersection of machine learning, large-scale distributed systems, and AI safety, developing the platforms and tools that enable our safeguards to operate reliably at scale.

Fireworks is seeking a Member of Technical Staff on the Research team, designed for PhD candidates who want their work to reach production. This role involves pushing the boundaries of generative AI, advancing LLMs and multimodal systems through foundational research. You will enhance model efficiency, accuracy, and scalability, directly shaping high-performance AI infrastructure. You will be paired with a senior researcher as a mentor and given a real research problem from day one, collaborating with top experts in deep learning, distributed systems, and optimization. The work you contribute will be deployed by leading companies, often within weeks.

You will help define how machine learning models run across Cloudflare’s global network, from frontier open LLMs and real-time voice models to customer-deployed models served on heterogeneous GPUs and next-generation accelerators. You’ll work with systems engineers, product teams, hardware partners, and AI/ML engineers to bring models into production with low latency, strong reliability, and efficient resource use. This role combines applied ML, inference optimization, evaluation, and production engineering, with a focus on benchmarking models, improving serving performance, validating quality, and building tooling that helps Cloudflare and its customers ship AI applications at Internet scale.

Fiddler is seeking a Solutions Architect, Customer Success to ensure clients achieve significant, measurable outcomes from their AI observability investments. In this role, you will act as both a technical expert and a trusted advisor, connecting complex Machine Learning (ML) and Large Language Model (LLM) systems with tangible business value. By guiding customers through the onboarding process, building integrations, and advocating for their needs internally, you will help them deploy trustworthy AI at scale, contributing to Fiddler's growth through successful adoption, renewals, and expansion.

$160k - $200k

Cohere

Cohere is seeking a Staff Software Engineer to join our internal infrastructure team, responsible for building and operating world-class infrastructure and tools for training, evaluating, and serving Cohere's foundational AI models. You will work closely with AI researchers to support their AI workload needs on cutting-edge systems, focusing on stability, scalability, and observability. This role involves building and operating superclusters across multiple clouds, directly accelerating the development of industry-leading AI models. Participation in a 24x7 on-call rotation is required and compensated.

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.

Applied Intuition is seeking a Perception & Fusion Engineer to join their team in developing, integrating, and maintaining real-time sensor software solutions for autonomous vehicles across various domains. You will design, develop, and integrate sensor fusion algorithms to interface with different platforms and support autonomy team objectives. This role involves developing autonomous sensor controllers to fuse information from state-of-the-art sensor systems like EO, IR, acoustics, radar, and RF.

$10k - $20k

World Labs is a frontier AI research and product company focused on spatial intelligence, advancing beyond large language models. Co-founded by leading researchers, the company is pioneering world models that perceive, generate, reason, and interact with virtual and physical worlds. Their flagship product, Marble, transforms various media into navigable 3D worlds, with applications in gaming, film, architecture, robotics, and immersive experiences. Backed by significant investment, World Labs is building a world-class team at the intersection of AI research and real-world deployment.

$250k - $325k

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.

Sarvam is building India's sovereign AI platform, focusing on research, models, infrastructure, and applications to make AI work for India. The company partners with leading enterprises and public institutions. This role focuses on Sarvam's consumer and enterprise products, which are native applications running on devices like Android phones, iPhones, Windows laptops, Linux desktops, and Macs. These applications require on-device inference, audio capture/streaming, and offline functionality, needing to perform reliably across a wide range of hardware and network conditions in India. The Staff Engineer will own the client application layer across mobile and desktop, using Kotlin Multiplatform (KMP) as a unifying technology to manage development across five platforms without separate codebases. This is an individual contributor role where you will collaborate closely with backend, model serving, and product teams.

OpenAI

OpenAI is seeking Machine Learning Engineers to join its API Multicloud team, focusing on extending OpenAI's API platform into strategic cloud environments, starting with AWS. This role involves building and improving AI systems that help strategic partners adapt OpenAI models for cloud-native use cases. You will operate at the intersection of production ML systems, developer platforms, model behavior, and large-scale infrastructure, spanning post-training workflows, evaluation, data pipelines, and API/infrastructure integration. The ideal candidate will enjoy working with external technical partners, diagnosing issues, and translating learnings into platform improvements, collaborating closely with Research, Applied, Safety Systems, and infrastructure teams.

OpenAI

OpenAI is seeking a senior Machine Learning Data Scientist to lead forecasting initiatives within the Strategic Finance team. This role involves building and scaling robust, interpretable, and production-ready forecasting systems to predict key business metrics like user growth, revenue, and compute consumption. You will be a founding member of the Forecasting pillar, collaborating closely with product managers, researchers, engineers, and finance leaders to operationalize insights, influence company strategy, and build foundational forecasting capabilities. This is a highly cross-functional position requiring technical excellence, product intuition, and business acumen.

Fireworks is seeking a Member of Technical Staff for its Research team to push the boundaries of generative AI. This role involves advancing LLMs and multimodal systems through foundational research, focusing on enhancing model efficiency, accuracy, and scalability to shape high-performance AI infrastructure. You will collaborate with experts in deep learning, distributed systems, and optimization to translate cutting-edge research into practical applications and influence how leading companies build and deploy AI.

OpenAI

As a Researcher for cybersecurity risks, you will help design and implement an end-to-end mitigation stack to reduce severe cyber misuse across OpenAI’s products. This role requires strong technical depth and close cross-functional collaboration to ensure safeguards are enforceable, scalable, and effective. You’ll contribute directly to building protections that remain robust as products, model capabilities, and attacker behaviors evolve. Models are becoming increasingly capable—moving from tools that assist humans to agents that can plan, execute, and adapt in the real world. As we push toward AGI, cybersecurity becomes one of the most important and urgent frontiers: the same systems that can accelerate productivity can also accelerate exploitation.

Fiddler is building trust into AI, especially with the rise of Generative AI and Agents. Our platform helps organizations deploy trustworthy and transparent AI solutions by monitoring, evaluating, securing, analyzing, and improving AI applications. We partner with AI-first organizations to establish responsible AI practices, enabling engineering and business teams to understand AI outcomes. Joining Fiddler means making an impact by ensuring AI applications at production scale have operational transparency and security, offering monumental learning opportunities in a rapidly innovating industry.

$174k - $215000k

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.

Applied Intuition is seeking a Perception Autonomy Engineer to develop, integrate, and maintain real-time sensor software solutions for autonomous vehicles across various domains. You will work with a team to enhance capabilities and demonstrate solutions in real-world scenarios on diverse hardware platforms. The role involves designing, developing, and integrating AI/ML sensor algorithms to interface with different platforms, ingest mission information, and support autonomy objectives. This includes fusing information from state-of-the-art sensors like EO, IR, acoustics, radar, and RF to create autonomous sensor controllers.

$10k - $20k

Skild AI is seeking a Computer Vision AI & ML Engineer to design, build, and deploy advanced perception systems for real-world robotics and automation. You will work across the full machine learning lifecycle—model development, data strategy, evaluation, and production integration—to deliver robust, high-performance vision capabilities. This role combines applied research with hands-on engineering and offers the opportunity to influence both architecture and roadmap decisions.

OpenAI

As a Machine Learning Engineer in OpenAI's Integrity team, you will work with state-of-the-art models and classifiers, experiment with new architectures, and advance our capabilities in content and user understanding. This role offers the chance to transform research breakthroughs into tangible solutions that enhance platform trust and safety, particularly for those excited about training LLMs and building ML models. You will be instrumental in designing and deploying advanced machine learning models to solve real-world problems, bringing AI-driven applications from concept to implementation. Collaboration with researchers, software engineers, and product managers is key to delivering AI-powered solutions for complex business challenges. You will also optimize and scale data pipelines, ensure models are production-ready, and contribute to projects requiring cutting-edge technology. Staying ahead of the curve in machine learning and AI developments, participating in code reviews, and leading by example are essential aspects of this role, as is monitoring and maintaining deployed models to ensure continued value delivery.

Skild AI is seeking a Software Engineer to develop and optimize the software infrastructure and tools for training cutting-edge AI models. This role involves building scalable and efficient training pipelines and frameworks that support the entire machine learning lifecycle, from data preparation to model deployment. You will collaborate with researchers and machine learning engineers to integrate new algorithms and techniques, pushing the boundaries of AI in real-world robotics. The position also focuses on exploring efficient data utilization within training pipelines.