JAX Jobs in San Jose, CA

21 - 40 of 60 Jobs

Senior Research Engineer - Foundation Models, Ads IntegrityJob Title - Senior Research Engineer - Foundation Models, Ads Integrity

TikTokCompany Name - TikTok

•

San Jose, California, USA

Full-time

Location : San Jose Employment Type : Regular Job Code : A177519 Apply to this job Share this listing: Responsibilities Our Business Integrity team has a strong user focus and a dedication to technical excellence. We aim to meet our users' needs with reliable and high-performing platforms and services. We are looking for strong machine learning engineers who are excited to grow their business understanding, build highly scalable machine learning models, and partner across disciplines with

Software Engineer, Machine Learning, CalibrationJob Title - Software Engineer, Machine Learning, Calibration

Nuro Inc.Company Name - Nuro Inc.

•

Mountain View, California, USA

Full-time

Who We Are Nuro exists to better everyday life through robotics. Founded in 2016, Nuro has spent eight years developing autonomous driving (AD) technology and commercializing AD applications. The Nuro Driver is our world-class autonomous driving system that combines AD hardware with our generalized AI-first self-driving software. Built to learn and improve through data, the Nuro Driver is one of the few driverless autonomous technologies on public roads today. Nuro has raised over $2B in capi

Senior Software Engineer, Machine Learning, 3D ReconstructionJob Title - Senior Software Engineer, Machine Learning, 3D Reconstruction

Nuro Inc.Company Name - Nuro Inc.

•

Mountain View, California, USA

Full-time

Software Engineer, Machine Learning, 3D ReconstructionJob Title - Software Engineer, Machine Learning, 3D Reconstruction

Nuro Inc.Company Name - Nuro Inc.

•

Mountain View, California, USA

Full-time

Senior Full-Stack Software EngineerJob Title - Senior Full-Stack Software Engineer

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. Today, we're at the forefront of AI innovation powering breakthroughs in research, autonomous vehicles, robotics, and more. The DGX Cloud team builds and operates the AI infrastructure that fuels this progress. We're looking for a Senior Full-Stack Software Engineer to join our DGX Cloud AI Infrastructure team and help deliver the next-generation user experience for NVIDIA's GPU clusters

Senior Software Engineer, TensorRT InferenceJob Title - Senior Software Engineer, TensorRT Inference

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

We are now looking for a Senior Software Engineer for our TensorRT Inference team! At NVIDIA, we're at the forefront of innovation, driving advancements in AI and machine learning to solve some of the world's most challenging problems. We're seeking talented and motivated engineers to join our TensorRT team in developing the industry-leading deep learning inference software for NVIDIA AI accelerators. What you'll be doing: As a Senior Software Engineer in the TensorRT team, you will be respons

Senior Deep Learning Compiler Engineer - XLAJob Title - Senior Deep Learning Compiler Engineer - XLA

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. It's a unique legacy of innovation that's fueled by great technology-and amazing people. Today, we're tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, robots, and self-driving cars that can understand the world. Doing what's never been done before takes vision, innovation, and the world's best talent.

Sr. Manager Software Development, Efficient Generative AIJob Title - Sr. Manager Software Development, Efficient Generative AI

AMD (Advanced Micro Devices)Company Name - AMD (Advanced Micro Devices)

•

San Jose, California, USA

Full-time

WHAT YOU DO AT AMD CHANGES EVERYTHING We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences - the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our mission is the AMD culture. We push the limits of innovation to solve the world's most important challenges. We strive for execution excellenc

Senior Solutions Architect, GPU - Cloud Service ProvidersJob Title - Senior Solutions Architect, GPU - Cloud Service Providers

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

Join our team at NVIDIA and help bring AI solutions to our largest customers. We are seeking an expert Solutions Architect to assist customers in building AI/ML and HPC software solutions at scale. As a member of our Solutions Architecture team, you will collaborate with strategic customers, providing end-to-end technology solutions and technical support based on our product strategy. Come join us! What you'll be doing: Working with tech giants to develop and demonstrate solutions based on NVID

Senior Software Engineer, Machine Learning, CalibrationJob Title - Senior Software Engineer, Machine Learning, Calibration

Nuro Inc.Company Name - Nuro Inc.

•

Mountain View, California, USA

Full-time

Senior Software Engineer, Machine Learning, Sensor SimulationJob Title - Senior Software Engineer, Machine Learning, Sensor Simulation

Nuro Inc.Company Name - Nuro Inc.

•

Mountain View, California, USA

Full-time

Senior Software Engineer, Generative-AIJob Title - Senior Software Engineer, Generative-AI

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

At NVIDIA, we're not just building the future-we're generating it. Our Cosmos generative AI engineering team is pushing the boundaries of what's possible across multimodal learning, video generation, synthetic data, and intelligent simulation. We welcome hard-working engineers and applied scientists with deep experience in generative modeling to help define the next era of AI computing. What you'll be doing: Design, post-train, and optimize novel world models (e.g., diffusion video models, VLMs

Senior DGX Cloud Performance EngineerJob Title - Senior DGX Cloud Performance Engineer

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

NVIDIA DGX Cloud is an end-to-end, scalable AI platform for developers, offering scalable capacity built on the latest NVIDIA architecture and co-engineered with the world's leading cloud service providers (CSPs). We are seeking highly skilled Parallel and Distributed Systems engineers to drive the performance analysis, optimization, and modeling to define the architecture and design of NVIDIA's DGX Cloud clusters. The ideal candidate will have a deep understanding of the methodology to conduct

Staff Engineer, AI/ML Software CompilerJob Title - Staff Engineer, AI/ML Software Compiler

Samsung Electronics AmericaCompany Name - Samsung Electronics America

•

San Jose, California, USA

Full-time

Please Note: To provide the best candidate experience amidst our high application volumes, each candidate is limited to 10 applications across all open jobs within a 6-month period. Advancing the World's Technology Together Our technology solutions power the tools you use every day--including smartphones, electric vehicles, hyperscale data centers, IoT devices, and so much more. Here, you'll have an opportunity to be part of a global leader whose innovative designs are pushing the boundaries o

Senior DGX AI Cloud Performance Analysis Tools EngineerJob Title - Senior DGX AI Cloud Performance Analysis Tools Engineer

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

Joining NVIDIA's DGX Cloud AI Efficiency Team means contributing to the infrastructure that powers our innovative AI research. This team focuses on optimizing efficiency and resiliency of AI workloads, as well as developing scalable AI and Data infrastructure tools and services. Our objective is to deliver a stable, scalable environment for AI researchers, providing them with the necessary resources and scale to foster innovation. We are seeking excellent Software Engineers to design and develop

Principal Generative-AI Software EngineerJob Title - Principal Generative-AI Software Engineer

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

Senior/Staff Software Engineer, ML Infrastructure, OptimizationJob Title - Senior/Staff Software Engineer, ML Infrastructure, Optimization

Nuro Inc.Company Name - Nuro Inc.

•

Mountain View, California, USA

Full-time

Senior AI Software Engineer, GenAI FrameworkJob Title - Senior AI Software Engineer, GenAI Framework

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

We are now looking for AI Software Engineers for NeMo! NVIDIA NeMo is an open-source, scalable and cloud-native framework built for researchers and developers working on Large Language Models (LLM), Multimodal (MM), and Speech AI. NeMo provides end-to-end model training, including data curation, alignment, customization, evaluation, deployment and tooling to optimize performance and user experience. In this critical role, you will expand NeMo Framework's capabilities, enabling users to develop,

Machine Learning Engineer Intern (TikTok - P- LLM/Content Understanding) - 2025 Start (PhD)Job Title - Machine Learning Engineer Intern (TikTok - P- LLM/Content Understanding) - 2025 Start (PhD)

TikTokCompany Name - TikTok

•

San Jose, California, USA

Full-time

Location : San Jose Employment Type : Intern Job Code : A45454 Apply to this job Share this listing: Responsibilities Our mission is to empower professional creators, media partners, and organizations to produce high-quality content that engages, entertains, and informs TikTok's diverse global audience. We are a product-facing team and you will work closely with a multi-disciplinary team (Product, Engineers, Design, and Data) to define and build solutions for creators who inspire members

Principal AI/ML Engineer - Onboard Embodied AIJob Title - Principal AI/ML Engineer - Onboard Embodied AI

General MotorsCompany Name - General Motors

•

Mountain View, California, USA

Full-time

Job Description Hybrid: This role is categorized as hybrid. This means the successful candidate is expected to report to the Mountain View Technical Center in the Bay Area three times per week, at minimum. Role: As a Technical Lead in Machine Learning within the Onboard Embodied AI organization, you will be a senior individual contributor driving cutting-edge end-to-end machine learning solutions directly impacting autonomous driving performance. Your role is pivotal in designing, architecting