TensorFlow Jobs

85 open roles mentioning TensorFlow

AI Performance Engineer - New Grad

9d ago
S

SambaNova

As a new grad AI Performance Engineer, you will be responsible for bringing up new open-source foundation models on the SambaNova dataflow platform and optimizing their performance. This involves profiling model execution, identifying performance bottlenecks across compiler, runtime, and hardware layers, and applying optimizations such as quantization, kernel fusion, and improved scheduling. You will collaborate with machine learning, compiler, runtime, and hardware teams to integrate performance improvements into SambaNova products and benchmark throughput and latency to track changes across releases.

$131k - $197k

Austin, Texas, United States; San Jose, California, United States onsite
Meta LlamaPythonPyTorch +8 more

Applied Research Intern, Integrated Sensing and Communications (ISAC)-NVIDIA Aerial - Summer 2027

10d ago
n

nvidia

NVIDIA is seeking a graduate research intern to advance integrated sensing and communications (ISAC) on GPU-accelerated platforms. The intern will join a team building AI-native wireless systems and develop, train, and test ISAC algorithms end-to-end on the NVIDIA Aerial Testbed. This role is ideal for a student passionate about deep algorithm work and translating ideas into functional systems.

$0k - $0k

US, CA, Santa Clara remote
PythonPyTorchTensorFlow +5 more

Research, Tinker, RL Systems

10d ago
T

Thinking Machines Lab

Thinking Machines is building AI that extends human will and judgment. We are developing Tinker, a fine-tuning API that empowers researchers and developers to customize frontier AI models. This role focuses on building and enhancing our training systems for Tinker, specifically involving RL systems, numerics, and kernels, to enable users to train models with their own data and algorithms.

$350k - $475k

San Francisco onsite FullTime
PythonLLMFine-Tuning +7 more

Research Scientist (diffusion)

10d ago
G

Genmo

Genmo is a research lab focused on developing advanced video world models to understand, simulate, and interact with the physical world, aiming to unlock the right brain of AGI. We are seeking an exceptional Research Scientist to pioneer cutting-edge diffusion models for text-to-video generation. This role involves creating novel architectures and algorithms to transform written descriptions into high-quality, coherent video content, pushing the boundaries of physical intelligence and enabling robots to learn and act in dynamic environments.

San Francisco HQ onsite FullTime
PythonPyTorchTensorFlow +7 more

Senior Machine Learning Engineer

10d ago
C

Cresta

Cresta is seeking a Senior Machine Learning Engineer to join their team and work across several high-impact AI initiatives. The role involves leading and building next-generation agentic AI systems that augment contact center agents in real time, designing evaluation frameworks to improve the reliability and performance of LLM-powered agents, and architecting/scaling LLM and retrieval-augmented generation pipelines. Final team placement will be determined by experience, strengths, and business needs.

$205k - $270k

United States (Remote) remote
LLMRAGAI Agents +8 more

Staff Machine Learning Engineer

10d ago
C

Cresta

Cresta is seeking a Staff Machine Learning Engineer to work on high-impact AI initiatives, focusing on next-generation agentic AI systems that augment contact center agents in real-time. This role involves leading and building these systems, requiring strong ML foundations, deep expertise in LLMs and prompting techniques, and a rapid prototyping mindset to translate research into scalable, production-grade systems. The engineer will also focus on improving the reliability, robustness, and performance of LLM-powered agents by designing evaluation frameworks, diagnosing failure modes, and defining measurable quality metrics. Additionally, the role includes architecting and scaling LLM and retrieval-augmented generation pipelines to process complex data and deliver real-time intelligence.

$230k - $300k

United States (Remote) onsite
LLMRAGAI Agents +4 more

Senior Perception Software Engineer

14d ago
n

nvidia

NVIDIA is seeking a passionate 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

US, CA, Remote onsite
PythonPyTorchTensorFlow +5 more

Applied Scientist

14d ago
B

Blee

Blee is seeking an Applied Scientist to specialize in agent system design, focusing on architecting and implementing the next generation of intelligent compliance systems. This role involves designing frameworks where multiple agents, powered by custom and foundational models, collaborate to identify, track, and explain compliance flags. You will also be responsible for building and optimizing applied AI/ML pipelines, including model fine-tuning, prompt engineering, evaluation frameworks, and customer-specific AI configurations, and training models based on user feedback to deliver more targeted compliance recommendations.

$150k - $275k

San Francisco hybrid FullTime
PythonLLMRAG +8 more

Senior GPU Functional Modeling Architect

16d ago
n

nvidia

NVIDIA is seeking Architects, Functional Modeling Engineers, and Simulation experts to join its Architecture Modeling group. In this role, you will collaborate with world-class architects to enable functional simulation platforms across GPU generations. You will be instrumental in modeling GPU architecture, baseboard components, and other features, working within a matrixed environment to document, design, develop tools for analysis and simulation, and validate models. The position involves familiarizing yourself with various functional and performance simulation models across NVIDIA, developing tests and testing infrastructure for new architectures, and guiding the improvement of the simulation platform to support future GPU architectures. You will also have the opportunity to develop or use AI to assist with daily tasks.

$152k - $288k

US, CA, Santa Clara remote
LangChainPythonPyTorch +8 more

NVIDIA 2027 Internships: Deep Learning

18d ago
n

nvidia

NVIDIA is seeking strategic, ambitious, hard-working, and creative individuals for its 2027 Deep Learning internship program. This 12-week full-time internship offers an excellent opportunity to expand your career and gain hands-on experience with industry-leading Deep Learning teams. Interns will work on projects with a measurable impact on NVIDIA's business, contributing to advancements in AI and digital twins that are transforming major industries. We are looking for students pursuing a B.S., M.S., or Ph.D. degree in a relevant field who are passionate about tackling complex challenges.

$0k - $0k

US, CA, Santa Clara remote
AWSDockerKubernetes +9 more

PhD Research Intern, Quantum Simulation and AI - 2027

18d ago
n

nvidia

NVIDIA Research is seeking leading researchers in the area of simulation and AI for quantum. You should have a strong background of research excellence, software systems building experience, and interest in advancing the state-of-the-art in quantum computing and applications of GPUs to the quantum computing ecosystem. Specific areas of research interest include but are not limited to bespoke quantum representations, tensor network methods, quantum surrogate modeling, quantum inverse design, and automated proving for quantum systems. This role involves researching and developing creative and innovative algorithms and software systems for the quantum computing ecosystem, collaborating with research and product teams, and publishing original research.

$0k - $0k

US, CA, Santa Clara remote
PythonPyTorchTensorFlow +5 more

PhD Research Intern, Generalist Embodied Agents Research - 2027

18d ago
n

nvidia

NVIDIA is seeking a research intern to contribute to building humanoid robot foundation models and systems within the Generalist Embodied Agent Research (GEAR) group. The mission is to create general-purpose embodied agents capable of learning and mastering complex skills in both virtual and physical environments. You will collaborate with a research team known for influential work in multimodal foundation models, large-scale robot learning, game AI, and physical simulation, contributing to significant research projects and product roadmaps, including recent milestones like Project GR00T.

$0k - $0k

US, CA, Santa Clara remote
PythonLLMPyTorch +9 more

NVIDIA 2027 Internships: Autonomous Vehicles and Robotics

18d ago
n

nvidia

NVIDIA is seeking strategic, ambitious, hard-working, and creative individuals for their 2027 Autonomous Vehicles and Robotics internships. This 12-week full-time program offers an excellent opportunity to expand your career and gain hands-on experience with industry-leading teams. Interns will work on projects with a measurable business impact, contributing to NVIDIA's pioneering work in AI and digital twins that is transforming major industries.

$0k - $0k

US, CA, Santa Clara onsite
PythonPyTorchTensorFlow +10 more

NVIDIA Spring 2027 Internships: Developer and Performance Technology

18d ago
n

nvidia

NVIDIA is seeking ambitious and creative individuals for its Spring 2027 Developer and Performance Technology internships. These 8-12 month full-time internships offer a unique opportunity to gain hands-on experience with industry-leading Deep Learning teams, working on projects that have a measurable impact on NVIDIA's business. Interns will tackle complex challenges in areas such as AI, digital twins, gaming, robotics, self-driving cars, and virtual worlds. The program is designed for students pursuing a B.S. or M.S. degree in a relevant field, providing a chance to expand their careers and contribute to transformative technologies.

$0k - $0k

US, CA, Santa Clara remote
AWSAzureGoogle Cloud +13 more

Senior Software Engineer, JAX

19d ago
n

nvidia

NVIDIA is seeking senior engineers to enhance its AI platform, focusing on performance optimizations within deep learning frameworks like JAX. The goal is to create a robust, efficient, and scalable platform for data handling, training, and analysis across various deep learning solutions. This role requires strong programming and system design skills, along with excellent communication and planning abilities, to contribute to the next generation of numerical computing and machine learning research tools.

$184k - $357k

US, CA, Santa Clara onsite
PythonPyTorchTensorFlow +7 more

Applied Machine Learning Engineer

20d ago
P

PermitFlow

PermitFlow is seeking an Applied Machine Learning Engineer to develop the ML foundation for their AI agents, which are redefining how America builds by tackling complex information challenges in the construction industry. This role involves designing, prototyping, and deploying intelligent systems for document processing, insight extraction, and autonomous permitting workflows. You will own the end-to-end ML lifecycle, from research and data engineering to production deployment and continuous evaluation, contributing to the AI infrastructure that accelerates housing, clean-energy, and infrastructure development.

$175k - $250k

New York City, NY hybrid FullTime
OpenAIAnthropicPinecone +9 more

Machine Learning Engineer

21d ago
H

Handshake AI

Handshake is seeking a Machine Learning Engineer I for the Growth Relevance team to help shape the evolution of careers in the AI economy. This role will focus on developing, deploying, and enhancing machine learning systems for lifecycle optimization, personalized notifications, and monetization strategies. You will work with cutting-edge ML infrastructure, including embedding-based retrieval, Graph Neural Networks, and multi-stage rankers, contributing to critical marketplace metrics and responsible AI practices.

$150k - $189k

San Francisco, CA remote FullTime
AWSAzureGoogle Cloud +9 more

Senior System Simulation Architect

23d ago
n

nvidia

NVIDIA is seeking a Senior System Simulation Architect to drive the development of CPU architectures for AI, deep learning, high-performance computing, gaming, virtual reality, and autonomous vehicles. This role involves developing and integrating full-system functional models, bringing up system and application software in simulation and emulation, and supporting CPU architects and performance engineers. The position is crucial for pushing performance boundaries for NVIDIA's CPU products and integrating them seamlessly with NVIDIA's broader technology stack to enable faster AI model training, agentic use-cases, efficient data processing, and scalable cloud deployments.

$184k - $357k

US, CA, Santa Clara onsite
PythonPyTorchTensorFlow +9 more

Senior Machine Learning Engineer, Perception - Autonomous Driving

24d ago
n

nvidia

Our team is building the machine learning backbone of the Perception component for NVIDIA DRIVE AV, focusing on solving complex problems for self-driving cars. We are seeking experienced Machine Learning Engineers with expertise in computer vision, LiDAR & camera perception, multi-sensor fusion, and AI infrastructure. The role involves inventing human-level AI for perception under real-world conditions, contributing to a defining technological advancement of our generation.

$184k - $357k

US, CA, Santa Clara onsite
PythonPyTorchTensorFlow +8 more

Machine Learning Architect – Automotive

24d ago
Accenture

Accenture

Accenture Industry X is seeking an experienced Machine Learning Architect to join their Automotive Engineering team. This role will contribute to the development of next-generation intelligent vehicle solutions by designing and implementing Machine Learning architectures for automotive software platforms. The team focuses on shaping a future-looking Software-Defined Vehicle Platform that unifies various operating systems into a powerful developer ecosystem, driving innovation in ADAS and Infotainment.

Cluj-Napoca remote
PyTorchTensorFlowMachine Learning +7 more

Get new AI jobs in your inbox

A weekly digest of the newest AI engineering roles.

© 2026 AI Job Board. All rights reserved.