JAX Jobs in California

Refine Results
21 - 40 of 73 Jobs

AI Research Scientist

Alldus

•

Santa Rosa, California, USA

Full-time

Job DescriptionMy Client is pioneering the future of Automation AI, developing an avant-garde platform that revolutionizes how repetitive tasks are identified and automated. Their system offers a groundbreaking "observe, learn, and automate" methodology, leveraging a foundational model to not only understand user behavior but to predict and automate mundane tasks without requiring any coding input. This AI-driven approach is designed to evolve continuously through human feedback, enhancing its c
Posted 15 days ago | Updated 1 hour ago

AI Research Scientist

Alldus

•

San Jose, California, USA

Full-time

Job DescriptionMy Client is pioneering the future of Automation AI, developing an avant-garde platform that revolutionizes how repetitive tasks are identified and automated. Their system offers a groundbreaking "observe, learn, and automate" methodology, leveraging a foundational model to not only understand user behavior but to predict and automate mundane tasks without requiring any coding input. This AI-driven approach is designed to evolve continuously through human feedback, enhancing its c
Posted 15 days ago | Updated 1 hour ago

Senior Machine Learning Engineer

Randstad - Torc

•

Remote

Contract

We are looking for a Senior Machine Learning Engineer to join our team and drive the development of advanced AI-driven solutions. You will be responsible for designing, deploying, and optimizing machine learning models that power real-world applications. This role requires deep expertise in model architecture, data engineering, MLOps, and production-scale AI systems. What you'll do:Design, develop, and deploy scalable, production-ready machine learning models for real-world applications.Architec
Posted 60+ days ago | Updated 9 days 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 10 days ago | Updated 1 hour ago

Developer Technologies Engineer, Robotics Reinforcement Learning

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

Today, NVIDIA is 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. As an NVIDIAN, you'll be immersed in a diverse, encouraging environment where everyone is inspired to do their best work. Come join the team and see how we can make a lasting impact on the world
Posted 24 days ago | Updated 1 hour 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 37 days ago | Updated 1 hour ago

Senior Compiler Verification Engineer - Deep Learning

NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

We are building the next generation of compiler technologies to accelerate deep learning workloads. We are looking for an engineer to work on compiler verification in the AI space. You will be solving critical problems working alongside a diverse set of minds in GPU computing and systems software, doing what you enjoy. If this sounds like a fun challenge, we want to hear from you! What you'll be doing: In this role you will work closely with compiler developers to verify new and state of the a
Posted 4 days ago | Updated 1 hour ago

Principal Deep Learning Software Engineer, LLM Performance

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are now looking for a Principal 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
Posted 45 days ago | Updated 1 hour ago

Staff Engineer, Machine Learning - Online Mapping

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 20 days ago | Updated 1 hour ago

Senior Deep Learning Software Engineer, Inference and Model Optimization

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

NVIDIA is at the forefront of the generative AI revolution! The Algorithmic Model Optimization Team specifically focuses on optimizing generative AI models such as large language models (LLM) and diffusion models for maximal inference efficiency using techniques ranging from neural architecture search and pruning to sparsity, quantization, and automated deployment strategies. Our work includes conducting applied research to improve model efficiency as well as developing an innovative software pl
Posted 37 days ago | Updated 1 hour ago

Software Developer, AI Networking

NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

NVIDIA is changing the world of AI Networking with groundbreaking technology. We are excited to be adding an AI Networking Software Developer to our AI Networking SW development and codesign team. In this role, you will be designing and implementing new protocols and algorithms that enable the best performance possible on NVIDIA networked supercomputers and data. centers. You will be delivering production grade solutions, getting hands on with creative and inventive technology, and collaborating
Posted 5 days ago | Updated 1 hour 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 15 days ago | Updated 1 hour ago

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 1 hour ago

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
Posted 60+ days ago | Updated 1 hour ago

Senior ML Software Engineer

NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

At NVIDIA, we're building the foundation for the next generation of Physical AI. Our Cosmos team is passionate about developing robust, production-grade open-source ML software to support machine learning and gen-AI research at scale. We're looking for expert-level Python engineers who are passionate about building production-ready systems and want to make a lasting impact through open-source contributions. If you care deeply about software craftsmanship, maintainability, and performance-and ha
Posted 39 days ago | Updated 1 hour ago

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
Posted 60+ days ago | Updated 1 hour ago

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
Posted 60+ days ago | Updated 1 hour 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 48 days ago | Updated 1 hour ago

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
Posted 45 days ago | Updated 1 hour 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 50 days ago | Updated 1 hour ago