jax Jobs in california

Refine Results
21 - 40 of 76 Jobs

Sr. Software Development Engineer - AI/ML Frameworks

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
Posted 8 days ago | Updated 9 hours ago

Software Engineer, Machine Learning, Calibration

Nuro Inc.

•

Mountain View, California, USA

Full-time

Who We Are Nuro is a self-driving technology company on a mission to make autonomy accessible to all. Founded in 2016, Nuro is building the world's most scalable driver, combining cutting-edge AI with automotive-grade hardware. Nuro licenses its core technology, the Nuro Driver , to support a wide range of applications, from robotaxis and commercial fleets to personally owned vehicles. With technology proven over years of self-driving deployments, Nuro gives the automakers and mobility platforms
Posted 60+ days ago | Updated 9 hours ago

Principal Software Engineer, TensorRT-LLM

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are now looking for a Principal Software Engineer, TensorRT-LLM ! NVIDIA is hiring experienced principal software engineer for its TensorRT-LLM team. Academic and commercial groups around the world are using GPUs to power a revolution in AI, enabling breakthroughs in areas like content creation, code generation and information synthesis, that have put AI on the precipice of an "iPhone moment". Join the team building the AI serving software which is foundational to product lines within NVIDIA
Posted 37 days ago | Updated 9 hours ago

Senior Deep Learning Software Engineer

NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

We are looking for a Senior Deep Learning Software Engineer to design and build our automated inference and deployment solution. As part of the team, you will be instrumental in defining a scalable architecture for DL inference with emphasis on ease-of-use and compute efficiency. Your work will span multiple layers of the DL deployment stack, encompassing developing features in high-level frameworks like PyTorch and JAX, designing and implementing a high-performance execution environment, low-le
Posted 42 days ago | Updated 9 hours ago

Senior GenAI Algorithms Engineer

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are now looking for a Senior Gen AI Algorithms Engineer! NVIDIA is seeking engineers to design, develop and optimize Artificial Intelligence solutions to diverse real-world problems. If you have a strong understanding of deep learning and in particular large language models and their multimodal variants, then this role may be a great fit for you! Collaborate and interact with internal partners, users, and members of the open source community to analyze, define and implement highly optimized A
Posted 60+ days ago | Updated 9 hours ago

Senior DGX Cloud AI Infrastructure Software Engineer

NVIDIA Corporation

•

Remote or 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 an AI infrastructure software engineer to join our
Posted 7 days ago | Updated 9 hours ago

Senior Software Engineer, Machine Learning, Sensor Simulation

Nuro Inc.

•

Mountain View, California, USA

Full-time

Who We Are Nuro is a self-driving technology company on a mission to make autonomy accessible to all. Founded in 2016, Nuro is building the world's most scalable driver, combining cutting-edge AI with automotive-grade hardware. Nuro licenses its core technology, the Nuro Driver , to support a wide range of applications, from robotaxis and commercial fleets to personally owned vehicles. With technology proven over years of self-driving deployments, Nuro gives the automakers and mobility platforms
Posted 60+ days ago | Updated 9 hours ago

Senior Software Engineer, Machine Learning, Calibration

Nuro Inc.

•

Mountain View, California, USA

Full-time

Who We Are Nuro is a self-driving technology company on a mission to make autonomy accessible to all. Founded in 2016, Nuro is building the world's most scalable driver, combining cutting-edge AI with automotive-grade hardware. Nuro licenses its core technology, the Nuro Driver , to support a wide range of applications, from robotaxis and commercial fleets to personally owned vehicles. With technology proven over years of self-driving deployments, Nuro gives the automakers and mobility platforms
Posted 60+ days ago | Updated 9 hours ago

Machine Learning Kernel Performance Engineer, Dojo

Tesla Motors

•

Palo Alto, California, USA

Full-time

As a member of the Dojo Machine Learning software team, you will be responsible for developing and optimizing machine learning kernels to run on a massively parallel machine. The ideal candidate will have a strong background in kernel development and performance optimization for AI workloads, with a passion for delivering high-performance implementations, which are simple to use. Responsibilities Develop and validate datapath kernels on a massively parallel machineDesign and deliver implementat
Posted 36 days ago | Updated 9 hours ago

Senior Research Engineer - Foundation Models, Ads Integrity

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
Posted 18 days ago | Updated 9 hours ago

Senior Machine Learning Engineer (Modeling), Support

Block Inc

•

California, USA

Full-time

Block is one company built from many blocks, all united by the same purpose of economic empowerment. The blocks that form our foundational teams - People, Finance, Counsel, Hardware, Information Security, Platform Infrastructure Engineering, and more - provide support and guidance at the corporate level. They work across business groups and around the globe, spanning time zones and disciplines to develop inclusive People policies, forecast finances, give legal counsel, safeguard systems, nurture
Posted 24 days ago | Updated 9 hours ago

Senior DGX AI Cloud Performance Analysis Tools Engineer

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
Posted 60+ days ago | Updated 9 hours ago

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
Posted 56 days ago | Updated 9 hours ago

Senior Software Architect, Always-On Profiling

NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

Are you ready to innovate GPU performance analysis for Machine Learning workloads?! Join our Developer Tools Always-On Profiling (AON) team as a Senior Software Architect, where you'll be pivotal in designing, implementing, and leading our Always-On Profiling service. This role demands deep technical expertise, a proven track record to solve ambiguous challenges, and strong technical leadership skills. What you'll be doing: Architect and Build Scalable Systems: Drive the design and implementati
Posted 20 hours ago | Updated 9 hours ago

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.
Posted 44 days ago | Updated 9 hours ago

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
Posted 60+ days ago | Updated 9 hours ago

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
Posted 60+ days ago | Updated 9 hours ago

Senior/Staff Software Engineer, ML Infrastructure, Optimization

Nuro Inc.

•

Mountain View, California, USA

Full-time

Who We Are Nuro is a self-driving technology company on a mission to make autonomy accessible to all. Founded in 2016, Nuro is building the world's most scalable driver, combining cutting-edge AI with automotive-grade hardware. Nuro licenses its core technology, the Nuro Driver , to support a wide range of applications, from robotaxis and commercial fleets to personally owned vehicles. With technology proven over years of self-driving deployments, Nuro gives the automakers and mobility platforms
Posted 60+ days ago | Updated 9 hours ago

Senior Full-Stack Software Engineer

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
Posted 16 days ago | Updated 9 hours ago

Senior AI Software Engineer, GenAI Framework

NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

NVIDIA is now looking for AI Software Engineers for ourGenAI Frameworks (Megatron Core and NeMo Framework) team. Megatron Core and NeMo Framework are open-source, scalable and cloud-native frameworks built for researchers and developers working on Large Language Models (LLM) and Multimodal (MM) foundation model pretraining and post-training. Our GenAI Frameworks provide end-to-end model training, including pretraining, alignment, customization, evaluation, deployment and tooling to optimize perf
Posted 60+ days ago | Updated 9 hours ago