TensorFlow Jobs

Anduril Industries is seeking a Machine Learning Engineer to join the Counter Intrusion MSE team. This role will own the machine learning stack for the team, focusing on designing and training multi-sensor object detection models for edge compute devices and developing learning algorithms to optimize autonomous systems. You will be responsible for the end-to-end design, implementation, and performance of the ML stack, including data collection and training infrastructure. Your work will directly contribute to Anduril's mission of delivering cutting-edge defense technology and will offer opportunities to identify and develop novel ML applications across the product portfolio.

$220k - $292k

We are seeking a Staff ML Tech Lead Manager to apply the latest research to solve our toughest problems. This role will be responsible for owning the entire machine learning stack for the Counter Intrusion team. You will design and train multi-sensor object detection models for perception on edge compute devices that push the boundaries of what's possible with our sensors. You will develop learning algorithms to optimize the behavior of autonomous systems. Your work will directly impact Anduril's ability to deliver cutting-edge defense technology, with the opportunity to identify and develop novel ML applications across our product portfolio.

$220k - $330k

Figma is seeking PhD interns to join the AI Research team, focusing on the intersection of AI and Design. You will develop innovative AI solutions for design and creative products through fundamental and applied research, contributing to new AI-driven features. The team works on optimizing agentic systems, building robust evaluators, and post-training models, tackling challenging technical problems in AI and creativity. This internship offers the opportunity to publish research in leading AI/ML conferences.

$0k - $0k

Figma is seeking a versatile and experienced Machine Learning / AI Engineer to join our growing AI team, working at the intersection of applied machine learning, infrastructure, and product innovation. Whether you’re building intelligent search systems, crafting scalable data pipelines, or enhancing AI-powered creativity tools, your work will drive user productivity, shape new product experiences, and advance the state of AI at Figma. You’ll collaborate closely with engineers, researchers, designers, and product managers across multiple teams to deliver high-quality ML-driven features and infrastructure. This is a high-impact, cross-functional role where you’ll shape both foundational systems and user-facing capabilities. This is a full time role that can be held from one of our US hubs or remotely in the United States.

$153k - $376k

We are the GPU Communications Libraries and Networking team at NVIDIA, responsible for delivering critical communication libraries like NCCL, NVSHMEM, and UCX for Deep Learning and High-Performance Computing (HPC) applications. These applications demand immense compute power and operate at scales involving tens of thousands of GPUs, connected via high-speed interconnects and networking. Communication performance is paramount, directly impacting end-to-end application performance, especially at massive scales. We are seeking a technical leader to manage our NVSHMEM and UCX libraries, offering an exceptional opportunity to advance the state-of-the-art and develop groundbreaking platforms.

$184k - $357k

Join a team that analyzes large-scale datacenter workloads on GPU-accelerated clusters. You will transform telemetry and workload data into clear findings and visuals, partnering with OS, container, GPU, and systems engineers. When useful, you will apply machine learning and deep learning techniques for categorization and forecasting, coordinating these into tools the team uses. This role involves analyzing signals to identify opportunities for application and platform improvements, working with high-dimensional data to spot trends, and communicating conclusions to engineers and leadership.

$152k - $242k

Accenture

We are seeking a motivated Junior Data and AI Engineer to join our technology team. This role is ideal for a recent graduate eager to build a career in data engineering, artificial intelligence, machine learning, and cloud-based analytics. You will support the design, development, and deployment of data-driven solutions to enhance decision-making, automate processes, and create intelligent digital products. Working alongside experienced engineers and business teams, you will learn industry best practices while contributing to impactful projects.

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.

$194k - $266k

NVIDIA is seeking a motivated Performance Engineer to join our GPU Communications Libraries and Networking team. This role involves influencing the roadmap of communication libraries that are critical for Deep Learning and High Performance Computing applications running on massive GPU clusters. You will analyze and optimize communication performance between GPUs, which directly impacts end-to-end application performance at scale. This is an excellent opportunity to advance the state of the art in GPU communication and contribute to NVIDIA's innovative technologies.

$152k - $288k

NVIDIA is seeking a highly motivated senior software engineer to join its communication libraries and network software team. This role involves developing and maintaining software for complex heterogeneous computing systems that power disruptive products in High Performance Computing and Deep Learning. You will be instrumental in designing and implementing highly-optimized communication runtimes and system software that enables interactions among GPUs and with other system components, contributing to the advancement of AI and HPC.

$152k - $288k

NVIDIA is seeking a Senior Software Architect to join our GPU Communications Libraries and Networking team. This role involves co-designing next-generation data center platforms and scalable communication software crucial for advancing Deep Learning and High Performance Computing (HPC). You will investigate and identify communication performance bottlenecks, design and implement new communication technologies, and explore innovative hardware and software solutions for future platforms. This is an opportunity to break performance barriers and deliver cutting-edge technologies that power AI and HPC workloads at massive scales.

$184k - $357k

Glean is seeking a Software Engineer to help build a scalable software platform, including enterprise search solutions. This role involves working across distributed systems, data pipelines, APIs, and user interfaces to deliver secure, high-quality products. You will contribute to a platform that powers intelligent search, an AI assistant, and AI agents, with a focus on enterprise-grade AI solutions. The work is deeply embedded in how people get things done, shipping agentic capabilities on an open, extensible stack.

$187k - $234k

Vectara is seeking a Platform Engineer to own the infrastructure that runs our deploy-anywhere platform. This role involves managing Kubernetes clusters for ML inference at scale, CI/CD pipelines, infrastructure as code, and the observability stack. You will be hands-on, writing Helm charts and Terraform, debugging issues, and shipping backend service features. The position requires deploying across AWS, GCP, and on-premises environments, including air-gapped setups, and participating in an on-call rotation to support enterprise customers.

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) is a unique and exclusive expert network that brings together researchers, academics, and domain leaders from over 70 fields to shape the future of AI. With over 90% of members holding doctorates, the collective focuses on how frontier AI systems are built, evaluated, and governed. Members participate in high-impact AI/ML projects, collaborate on research publications, and gain access to a vibrant community network. This role offers opportunities to work on projects such as building AI safety guardrails, optimizing PyTorch models, and evaluating ML code, with the potential to co-author technical reports and research papers.

Scale AI

The Machine Learning Engineer role at Scale focuses on applying advanced techniques in generative AI, computer vision, reinforcement learning, and agentic AI to enhance production environments for Scale's products and customer experience. Leveraging robust internal infrastructure and extensive datasets, ML engineers drive product improvements. The Public Sector team specifically deploys cutting-edge models for mission-critical government systems, working with large language models and computer vision. This includes developing agentic systems for complex challenges, building agent frameworks, and creating evaluation tools. Efforts also involve advancing reinforcement learning for LLMs and training advanced computer vision models for defense applications, including large-scale fine-tuning and developing generalizable foundation models.

$235k - $294k

Scale AI

Scale is seeking a Senior Machine Learning Engineer to own the application of generative AI, agentic AI, computer vision, and reinforcement learning to mission-critical production problems. In this role, you will tackle challenges without established solutions, proposing and building architectures, and ensuring their performance in customer-dependent environments. The Public Sector Machine Learning team focuses on deploying advanced models to government systems, with a primary emphasis on agentic systems built on large language models. This includes developing agents for complex operational and planning tasks, integrating retrieval pipelines and production APIs, managing state for long-running tasks, performing geospatial reasoning, and creating evaluation tooling. The role also involves applying reinforcement learning and computer vision for defense applications, including evaluation, labeling efficiency, and multimodal model training. As a Senior MLE, you will have design authority over a capability area, making final decisions on team patterns and ensuring they function under real-world constraints like classified environments, limited compute, and strict correctness requirements.

$235k - $294k

OpenAI

OpenAI is seeking a senior Machine Learning Data Scientist to lead its forecasting initiatives within the Strategic Finance team. This is a founding role for the Forecasting pillar, responsible for building and scaling robust, interpretable, and production-ready forecasting systems. The models developed will be crucial for predicting key business metrics like daily/weekly active users, revenue, lifetime value, compute consumption, and profitability, directly influencing critical business decisions and company strategy. This role requires a blend of technical expertise, product intuition, and business acumen, involving close collaboration with product managers, researchers, engineers, and finance leaders to operationalize insights and build foundational forecasting capabilities.

$347k - $445k

NVIDIA is seeking strategic, ambitious, hard-working, and creative individuals for its 2027 Deep Learning Computer Architecture internships. This 12-week full-time program offers an excellent opportunity to expand your career and gain hands-on experience with industry-leading Deep Learning Computer Architecture teams. Interns will work on projects with a measurable impact on NVIDIA's business, tackling complex challenges in AI and digital twins that are transforming major industries.

$0k - $0k

Ema builds AI Employees for HR, IT, and Finance, automating busy work across the employee experience. Founded by former executives from Google, Coinbase, Flipkart, and Okta, our team comprises engineers from premier tech companies and graduates of top universities. We are backed by leading investors and are at the forefront of Agentic AI, shipping real systems that run business processes at scale. We are seeking innovative and passionate Machine Learning Engineers who excel at solving complex problems with large datasets, transforming theoretical concepts into practical, scalable solutions, and pushing the boundaries of Natural Language Processing and Information Retrieval.

$160k - $220k