TensorFlow Jobs

SambaNova is seeking a Compiler Architect to join their team and drive innovation in compiler infrastructure and optimization algorithms. This critical role involves working with various layers of the compiler stack and coordinating with development teams to ensure state-of-the-art ML model performance on the SambaNova platform. Responsibilities include analyzing PyTorch and machine learning models to map operations onto the company's unique dataflow hardware.

$210k - $280k

SambaNova is seeking a talented and driven ML performance engineer to optimize and scale state-of-the-art foundation models on SambaNova's reconfigurable dataflow platform. You will work hands-on with advanced models to push the limits of throughput, latency, and efficiency. This role bridges the gap between deep learning and systems performance, collaborating across compiler, runtime, and hardware layers to deliver world-record performance for large-scale AI inference.

$180k - $255k

Joining NVIDIA's DGX Cloud AI Efficiency Team means contributing to the infrastructure that powers our innovative AI research. This team focuses on developing tools for optimizing efficiency and resiliency of AI workloads, including pre-training, post-training, and inference. Our objective is to deliver a stable, scalable environment for AI researchers, providing them with the necessary resources and scale to foster innovation. As a senior DGX Cloud AI Infrastructure software engineer, you will have the opportunity to work on innovative technologies that power the future of AI and data science and be part of a dynamic, diverse, and supportive team that values learning and growth.

$184k - $357k

NVIDIA is seeking an outstanding individual to join our platform SWQA team as a Platform Software Validation Engineer (SDET). This role involves developing and executing test plans for NVIDIA HGX/DGX/MGX platforms, covering servers, OS, firmware, and CUDA software stacks. You will be responsible for installing and testing various systems, OS, server firmware, and software stacks, and driving root cause analysis for reliability and validation test failures. The position requires building and developing server and OS level automation frameworks and tests, reviewing partner and supplier test results, and managing the bug lifecycle within an agile development team.

$140k - $270k

Join Tenstorrent’s AI Models team and work at the layer most ML engineers never see: bringing advanced models to life on custom AI hardware. You’ll own real workloads end‑to‑end including porting, tuning, and validating LLMs and vision models on our accelerator, and chasing down every last millisecond and percentage point of accuracy. This role is for people who love the craft of ML engineering and want their work to matter at silicon scale, not just behind another API. This role is hybrid, based in Cyprus. We welcome candidates at various experience levels for this role. During the interview process, candidates will be assessed for the appropriate level, and offers will align with that level, which may differ from the one in this posting.

Tenstorrent is seeking a Physical Design Engineer to lead cross-functional efforts in solving complex physical design challenges and developing end-to-end RTL-to-GDS methodologies across advanced nodes, with a strong focus on PPA and runtime improvements. The engineer will architect, integrate, and deploy AI/ML-driven solutions into production physical design flows, creating custom CAD tools and partnering with internal teams and EDA vendors to drive next-generation, ML-enabled capabilities. This role is hybrid, based out of Santa Clara, CA or Austin, TX or Fort Collins, CO. We welcome candidates at various experience levels for this role, and offers will align with the assessed level.

$100k - $500k

As an AI engineer at WRITER, you will be instrumental in shaping how enterprises leverage superintelligence. This role involves building tangible AI solutions that power the future of work for leading global companies. Your contributions will directly influence the performance, scalability, and ethical alignment of our advanced LLMs and AI agents, enabling businesses to achieve unprecedented productivity and innovation through AI grounded in their data.

$198k - $259k

Thinking Machines is building AI that extends human will and judgment, focusing on training frontier models, developing tools for model customization, and crafting interfaces for human-AI communication. This role is crucial for shaping the fundamental reasoning capabilities of our models by owning the late-stage training process. It involves a blend of fundamental research and practical engineering, making it ideal for individuals comfortable working at the intersection of pre-training and post-training.

$350k - $475k

OpenAI

The Monetization team is a new cross-functional group focused on building foundational systems to responsibly scale access to intelligence. Our mission is to develop user-first, privacy-preserving monetization products, including next-generation ads experiences, that strengthen user trust, unlock economic opportunity, and support OpenAI's long-term innovation. This role operates in a greenfield environment, moving quickly through prototyping, experimentation, and iterative deployment, partnering closely with Product, Design, and Research to bring research breakthroughs into real-world systems at global scale. As a Research Engineer, you will contribute to deploying state-of-the-art models in production environments, helping turn research breakthroughs into tangible solutions and making AI technology accessible and impactful.

$381k - $555k

Databricks

Databricks is seeking a Staff Backline Engineer to join their critical bridge team between Frontline Support and Engineering. This role focuses on handling complex technical issues and escalations within the Data and AI ecosystem, with a strong emphasis on customer success and proactive issue resolution. The engineer will be a technical expert and escalation point for challenging ML/AI issues, working collaboratively with Support, Engineering, Product, and customers to troubleshoot, identify root causes, and drive resolutions. The team prioritizes automation and tooling to enhance troubleshooting efficiency and improve platform supportability.

$0k - $0k

Databricks

The Applied AI team at Databricks is at the forefront of advancing GenAI-powered products, having launched Databricks Assistant, AI/BI Genie, and Agent Bricks. These products are used by hundreds of thousands of users daily, tackling challenging problems like code suggestion, text-to-sql generation, and knowledge QA. As GenAI products evolve, we are seeking engineers to drive the next phase of development, focusing on enhancing LLM quality, expanding GenAI capabilities, and strengthening platform architecture for AI interactions at scale.

$190k - $285k

NVIDIA is seeking a Senior Perception Software Engineer to design, develop, and productize autonomous driving solutions. As a member of the perception team, you will build world-class perception solutions, designing and implementing software components. You will be challenged to improve the robustness, efficiency, and reusability of these components, working closely with experienced perception engineers and researchers. This role involves executing the full software development lifecycle, writing well-designed, testable code, and integrating perception software into a functional system. You will also develop verification plans, document and maintain software functionality, and debug existing systems, ensuring alignment with project plans and industry standards. A key aspect of the role is deploying state-of-the-art perception DNN models to embedded platforms and analyzing system performance using benchmark data.

$224k - $357k

At Thinking Machines, we build tools that empower individuals to customize AI models for their specific needs, including the ability to train model weights. In this role, you will focus on frontier customization techniques and contribute to developing a leading post-training engine called Tinker. Your work will involve a comprehensive understanding of Reinforcement Learning (RL) science, and your findings will directly influence Tinker's training defaults, API design, and the open-source Tinker Cookbook. You will collaborate with internal research teams and contribute to open science initiatives for external partners.

$350k - $475k

Thinking Machines is building AI that extends human will and judgment, with a focus on training frontier models and developing tools that allow users to customize AI for their unique needs. This role leads the Fine-tuning Science team, focusing on frontier customization techniques and the Tinker post-training engine. It's a player-coach position where you will remain hands-on with scientific research while also guiding the team, shaping its direction, and ensuring research findings are integrated into Tinker. You will collaborate with internal research teams, contribute to open science, and engage with external users.

$350k - $475k

NVIDIA is seeking a Principal Systems Software Engineer for Semiconductor Inspection to join their team in Santa Clara. This role involves defining, building, and evaluating AI systems for semiconductor manufacturing, focusing on inspection and metrology. The work spans computer vision, time-series modeling, multimodal AI, and anomaly detection, with a strong emphasis on adapting and deploying these technologies within partner environments. The goal is to create integrated, reliable, and fast AI solutions that can be operated by fabs and equipment vendors, even in air-gapped facilities with limited data and tight production budgets.

$272k - $431k

NVIDIA is seeking versatile software engineers for its XLA team to build high-performance, production-grade software at the core of next-generation AI systems. This role involves developing compiler optimization algorithms for deep learning workloads, optimizing inference and training performance for the JAX framework and the OpenXLA compiler on NVIDIA GPUs at scale. You will collaborate with deep learning framework and hardware architecture teams to accelerate future AI software.

$152k - $288k

Genmo is a research lab focused on developing advanced video world models to understand, simulate, and interact with the physical world, aiming to advance physical intelligence and enable robots to learn and act. We are seeking an exceptional Software Engineer to join our research team in pushing the boundaries of visual generative AI. In this role, you will collaborate with experienced researchers to develop and enhance generative models, ensuring their safe and effective deployment. This is an opportunity to contribute to cutting-edge AI research in a fast-paced startup environment.

Parallel Wireless is seeking a skilled wireless systems engineer to lead the development of a multi-RAT digital twin for their Open RAN solution. This role involves executing production RAN software within a closed-loop simulation environment, enabling engineering teams to design, evaluate, and compare features for LTE, 5G NR, and 2G without the need for physical radio setups. The position is a senior individual contributor role at the intersection of wireless systems, simulation, production software, and AI/ML, focused on evolving an existing simulator into a scalable platform for feature development, testing, and optimization.

Anthropic

The Safeguards ML Inference Path team designs, builds, and operates the production infrastructure that powers Claude's ML-based safety systems. This role involves collaborating closely with safety researchers and inference engineers to bring new classifiers and novel classes of ML defenses to production, owning the research-to-production transfer of new safety technologies critical for every Claude model launch. The team builds for scale, serving thousands of ML classifiers for all requests on the token generation path and for every platform Claude runs on. Engineers will work at the intersection of machine learning, large-scale distributed systems, and AI safety, developing platforms and tools that enable safeguards to operate reliably at scale, with tooling and infrastructure used for every model launch.

$320k - $485k

Parallel Wireless is seeking experienced wireless algorithm experts to research and design advanced cellular communication algorithms for 5G and beyond products. This role will leverage neural networks and machine learning techniques to reimagine mobile networks with innovative, energy-efficient Open RAN solutions. The ideal candidate will contribute to the future of telecommunications by driving innovation through green and sustainable networks.