JAX Jobs in California

1 - 20 of 75 Jobs

Senior ML Compiler EngineerJob Title - Senior ML Compiler Engineer

General MotorsCompany Name - General Motors

•

Mountain View, California, USA

Full-time

Job Description Remote: This role is based remotely but if you live within a 50-mile radius of [Atlanta, Austin, Detroit, Warren, Milford or Mountain View], you are expected to report to that location three times a week, at minimum. Role: We are looking for a deep learning compiler engineer to build out our ML compiler for deploying machine learning models to a variety of ML hardware accelerators. You will develop and enhance GM's internal ML compiler for high performance, usability, and retar

Software Engineer, ML Inference Compiler & Deployment, GPU, CPUJob Title - Software Engineer, ML Inference Compiler & Deployment, GPU, CPU

Tesla MotorsCompany Name - Tesla Motors

•

Palo Alto, California, USA

Full-time

As a Software Engineer within our Autonomy teams, you will contribute to one of the most advanced and widely deployed AI Platforms in the world for Autopilot and our Humanoid Robot, Optimus. In this role, you will be responsible for the internal working of the AI inference stack and compiler running neural networks in millions of Tesla vehicles and Optimus. You will collaborate closely with the AI Engineers and Hardware Engineers to understand the full inference stack and design the compiler to

Software Engineer, ML Inference Compiler & Deployment, AI FrameworksJob Title - Software Engineer, ML Inference Compiler & Deployment, AI Frameworks

Tesla MotorsCompany Name - Tesla Motors

•

Palo Alto, California, USA

Full-time

As a Software Engineer within our Autonomy teams, you will contribute to one of the most advanced and widely deployed AI Platforms in the world, powering Autopilot and our Humanoid Robot, Optimus. In this role, you will be responsible for the internal working of the AI inference stack and compiler running neural networks in millions of Tesla vehicles and Optimus. You will collaborate closely with the AI Engineers and Hardware Engineers to understand the full inference stack and design the compi

Software Engineer, ML Inference Compiler & Deployment, AI AcceleratorJob Title - Software Engineer, ML Inference Compiler & Deployment, AI Accelerator

Tesla MotorsCompany Name - Tesla Motors

•

Palo Alto, California, USA

Full-time

Machine Learning Kernel Performance Engineer, DojoJob Title - Machine Learning Kernel Performance Engineer, Dojo

Tesla MotorsCompany Name - 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

Principal Machine Learning Engineer (NLP - AI Runtime Security)Job Title - Principal Machine Learning Engineer (NLP - AI Runtime Security)

PaloAlto NetworksCompany Name - PaloAlto Networks

•

Santa Clara, California, USA

Full-time

Company Description Our Mission At Palo Alto Networks everything starts and ends with our mission: Being the cybersecurity partner of choice, protecting our digital way of life. Our vision is a world where each day is safer and more secure than the one before. We are a company built on the foundation of challenging and disrupting the way things are done, and we're looking for innovators who are as committed to shaping the future of cybersecurity as we are. Who We Are We take our mission of

Staff Software Engineer, ML Inference Compiler & Deployment, OptimusJob Title - Staff Software Engineer, ML Inference Compiler & Deployment, Optimus

Tesla MotorsCompany Name - Tesla Motors

•

Palo Alto, California, USA

Full-time

As a Software Engineer within our Autonomy teams, you will contribute to one of the most advanced and widely deployed AI Platforms in the world for Autopilot and our Humanoid Robot, Optimus. In this role, you will be responsible for the internal working of the AI inference stack running neural networks in Optimus and millions of Tesla vehicles. You will collaborate closely with the Optimus AI Engineers and AI Hardware Engineers to understand the full inference stack, co-design models to fit the

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

Sr. Software Engineer TS/SCI PolygraphJob Title - Sr. Software Engineer TS/SCI Polygraph

LeidosCompany Name - Leidos

•

San Diego, California, USA

Full-time

Description Leidos has an exciting opportunity for a Sr. Software Engineer! *Must have an active TS/SCI Polygraph up front. No exceptions.* You will perform software development lifecycle (SDLC) activities as both an individual and a member of our top-notch agile development team building a large complex enterprise system. Development includes the full range of turning Agile user stories into implementable concepts, through development, unit testing, integration and test, and deployment of the

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

General MotorsCompany Name - General Motors

•

Mountain View, California, USA

Full-time

Principal AI Engineer, Enterprise AI PlatformJob Title - Principal AI Engineer, Enterprise AI Platform

PaloAlto NetworksCompany Name - PaloAlto Networks

•

Santa Clara, California, USA

Full-time

Deep Learning Vision Engineer, OptimusJob Title - Deep Learning Vision Engineer, Optimus

Tesla MotorsCompany Name - Tesla Motors

•

Palo Alto, California, USA

Full-time

Tesla is on a path to build humanoid robots at scale to automate repetitive and boring tasks. The goal of our Machine Learning team is to build and demonstrate a general robot learning system that can leverage AI to perform complex physical tasks, ranging from full body locomotion, and precise manipulation This specific role will do hands-on data analysis for Optimus's Machine Learning team, as well as build data pipelines, tools and applications to automate those analysis. This is a unique oppo

Machine Learning Systems EngineerJob Title - Machine Learning Systems Engineer

Signify TechnologyCompany Name - Signify Technology

•

San Jose, California, USA

Full-time

Job DescriptionMachine Learning Systems Engineer 3D Generative AI A well-funded and fast-growing company at the forefront of 3D generative AI is seeking a Machine Learning Systems Engineer to help build the most advanced end-to-end machine learning infrastructure for 3D content generation. This is a rare opportunity to work alongside a globally distributed team of experts in computer graphics, AI, and large-scale systems. With millions of users and tens of millions of 3D assets generated, the pl

AI Research ScientistJob Title - AI Research Scientist

AlldusCompany Name - Alldus

•

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