JAX Jobs in California

Refine Results
41 - 60 of 72 Jobs

Senior Software Engineer, TensorRT Inference

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 Software Engineer, Machine Learning, 3D Reconstruction

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 capit

Software Engineer, Machine Learning, 3D Reconstruction

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 capit

Senior Solutions Architect, GPU - Cloud Service Providers

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

Sr. Manager Software Development, Efficient Generative AI

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 Deep Learning Compiler Engineer - XLA

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.

Senior Software Engineer, Machine Learning, Calibration

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

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

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

Senior Deep Learning Software Engineer, LLM Performance

NVIDIA Corporation

Remote or Santa Clara, California, USA

Full-time

We are now looking for a Senior Deep Learning Software Engineer, LLM Performance! NVIDIA is seeking an experienced Deep Learning Engineer passionate about analyzing and improving the performance of LLM inference! NVIDIA is rapidly growing our research and development for Deep Learning Inference and is seeking excellent Software Engineers at all levels of expertise to join our team. Companies around the world are using NVIDIA GPUs to power a revolution in deep learning, enabling breakthroughs in

Senior Software Engineer, Generative-AI

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

Machine Learning Engineer, Efficiency Engineering - USDS

TikTok

San Jose, California, USA

Full-time

Location : San Jose Employment Type : Regular Job Code : A87619 Apply to this job Share this listing: Responsibilities The Efficiency Engineering team is all about our passion for crafting innovative tools and applications that empower IT operations and devops teams to achieve new levels of efficiency. We're a tight-knit crew of experienced developers, engineers and problem solvers fueled by a shared vision: streamlining operations, reducing manual workload, and empowering teams to do the

Senior AI Software Engineer, GenAI Framework

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,

Senior Software Engineer, Machine Learning, Sensor Simulation

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 capit

Principal Engineer, AI/ML Software Compiler

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

Principal Generative-AI Software Engineer

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, VLM

Senior DGX Cloud Performance Engineer

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

Principal Solutions Engineer, Infrastructure (SLURM & AI Focus)

AMD (Advanced Micro Devices)

Santa Clara, 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/Staff Software Engineer, ML Infrastructure, Optimization

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

Staff Engineer, AI/ML Software Compiler

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

Machine Learning Software Engineer (SMTS)

AMD (Advanced Micro Devices)

Santa Clara, 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 excellen