JAX Jobs in California

Refine Results
41 - 60 of 63 Jobs

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

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
Posted 59 days ago | Updated 3 hours ago

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
Posted 54 days ago | Updated 3 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 36 days ago | Updated 3 hours ago

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
Posted 33 days ago | Updated 3 hours ago

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

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
Posted 45 days ago | Updated 3 hours ago

AI System Research and Development Engineer - Optimization

Snowflake Inc.

•

Menlo Park, California, USA

Full-time

Where Data Does More. Join the Snowflake team. We are looking for talented System Developers and Researchers to join the Snowflake AI Research team and contribute to LLM inference and training system development, optimizations, and agentic systems. Our mission is to build the most efficient and scalable generative AI systems. Recent releases from our team include SwiftKV, an advanced inference optimization, and Arctic LLM, one of the largest open-source MoE foundation models. This is an exciti
Posted 60+ days ago | Updated 3 hours ago

AI and ML Infra Software Engineer, GPU Clusters

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 46 days ago | Updated 3 hours ago

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

Senior High-Performance LLM Training Engineer

NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

We are now looking for a Senior High-Performance LLM Training Engineer! NVIDIA is seeking experienced engineers specializing in performance analysis and optimization to improve the efficiency of LLM training workloads, which are shaping the world's most advanced computing systems. This position focuses on optimizing NVIDIA's high-performance LLM software stack in frameworks like PyTorch and JAX for high-performance training on thousands of GPUs, while also helping shape hardware roadmaps for th
Posted 60+ days ago | Updated 3 hours ago

ML Data Engineer - Healthcare Data Curation & Cleaning (1 Year Fixed Term)

Stanford University

•

Stanford, California, USA

Full-time

Stanford University is seeking a Big Data Architect 1 for a 1 year fixed term (possibility of renewal) to design and develop applications, test and build automation tools and support the development of Big Data architecture and analytical solutions. About Us: The Department of Biomedical Data Science merges the disciplines of biomedical informatics, biostatistics, computer science and advances in AI. The intersection of these disciplines is applied to precision health, leveraging data across th
Posted 33 days ago | Updated 3 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 54 days ago | Updated 3 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 54 days ago | Updated 3 hours ago

Staff Software Development Engineer - AI/ML frameworks, C/C++/Python

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 3 hours ago

Senior Math Libraries Engineers - Python APIs

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

NVIDIA is looking for a self-motivated and specialist software engineer for the design and development of high-performant Python APIs for our math libraries. In the last decade, Python has become the de-facto leading programming language for engineers in AI and data science, and more recently in HPC and scientific computing. NVIDIA has been at the forefront of providing GPU accelerated Deep Learning frameworks. These frameworks provide an efficient high-level programming interface allowing their
Posted 60+ days ago | Updated 3 hours ago

Lead Java Middle Tier Developer- C13 - JERSEY CITY

Citi

•

Remote or Jersey City, New Jersey, USA

Full-time

The Lead Java Middle Tier Developer is a senior level position responsible for establishing and implementing new or revised application systems and programs in coordination with the Technology team. The overall objective of this role is to lead applications systems analysis and programming activities. Responsibilities: Write good and well-tested code in JavaSOA API middleware implementation (both REST and SOAP)Write user/supporter documentation Development environment support (mostly automation
Posted 15 days ago | Updated 3 hours ago

AI Research Engineer, Foundation Models, Autonomy & Robotics

Tesla Motors

•

Palo Alto, California, USA

Full-time

Join us to help build the vision and language models that understand the world and act intelligently in it. We are hiring exceptional Software Engineers that can work across the machine learning stack, from designing and training neural architectures to deciding what data to collect and how to evaluate the model. You will join a small group of world-class deep learning experts to develop novel state-of-the-art neural networks and push the boundaries of AI research and development. Your work will
Posted 60+ days ago | Updated 3 hours ago

Software Engineer, Computer Vision

SpaceX

•

Hawthorne, California, USA

Full-time

SpaceX was founded under the belief that a future where humanity is out exploring the stars is fundamentally more exciting than one where we are not. Today SpaceX is actively developing the technologies to make this possible, with the ultimate goal of enabling human life on Mars. SOFTWARE ENGINEER, COMPUTER VISION We are looking for good people and driven, adaptable, resilient software engineers with experience developing software to accelerate quality, speed, and efficiency of manufacturing p
Posted 60+ days ago | Updated 3 hours ago

Software Engineer Staff

Lockheed Martin Corporation

•

Remote or Colorado Springs, Colorado, USA

Full-time

Job Description Space is a critical domain, connecting our technologies, our security and our humanity. While others view space as a destination, we see it as a realm of possibilities, where we can do more - we can innovate, invest, inspire and integrate our capabilities to transform the future. At Lockheed Martin Space, we aim to harness the full potential of space to cultivate innovation, reduce costs, and push the boundaries of what technology can achieve. We're creating future-ready solutio
Posted 46 days ago | Updated 3 hours ago