JAX jobs in California

Staff Software Engineer, ML Infra, Autonomy

Palo Alto, California

•

Today

About Rivian Rivian is on a mission to keep the world adventurous forever. This goes for the emissions-free Electric Adventure Vehicles we build, and the curious, courageous souls we seek to attract. As a company, we constantly challenge what's possible, never simply accepting what has always been done. We reframe old problems, seek new solutions and operate comfortably in areas that are unknown. Our backgrounds are diverse, but our team shares a love of the outdoors and a desire to protect it

Full-time

USD 206,500.00 - 258,100.00 per year

Principal AI Infrastructure & Accelerator Architect

Remote or New York, New York

•

Today

Are you a visionary technical leader passionate about defining the future of AI infrastructure and squeezing every drop of performance out of advanced hardware accelerators at scale? We are seeking a Principal Architect to lead, shape, and execute our technical strategy for AI performance, optimization, and hardware-software co-design. In this elite, highly visible role, you will define the architectural vision for both AI training and serving infrastructure, delivering massive industry-wide imp

Full-time

Principal AI Infrastructure & Accelerator Architect

Remote

•

Today

Full-time

LLM Application Engineer

San Francisco, California

•

Today

Software Guidance & Assistance, Inc., (SGA), is searching for an LLM Application Engineer for a Remote Full Time assignment with one of our premier Technology clients. Responsibilities : Design, develop, and deploy applications powered by large language models and intelligent workflow automation. Create agent-based systems that support reasoning, planning, memory management, tool integration, and multi-step task execution. Build orchestration frameworks that transform AI-generated outputs into d

Full-time

USD 120,000.00 - 205,000.00 per year

Remote Machine Learning Researcher

Remote

•

Today

Job Description A quantitative investment and research organization is seeking a Machine Learning Researcher / Engineer to join a highly collaborative, hands-on team. This is a fully remote opportunity with compensation up to $200,000 based on experience. This role is ideal for someone with a strong mathematical foundation who enjoys evaluating different machine learning approaches, transforming research into production-quality code, and delivering practical results. The team values intellectua

Easy Apply

Full-time

USD 200,000.00 per year

#Senior Embedded Software Engineer, Cloud Edge and Data Center Machine Learning

San Diego, California

•

Today

Company: Qualcomm Technologies, Inc. Job Area: Engineering Group, Engineering Group > Software Engineering General Summary: ** This position is not eligible for Qualcomm immigration sponsorship. ** Qualcomm's new Data Center Business Unit (BU) is looking for an experienced Senior Software Engineer to contribute to the design, development, and commercialization of software/firmware solutions for Cloud Edge and Data Center machine learning (ML) use cases running on premium-tier high-performa

Full-time

USD 111,300.00 - 166,900.00 per year

VP of Research, Machine Learning (ML), Artificial Intelligence (AI) Required, Work From Home - M

Remote

•

Today

VP of Research, Machine Learning (ML), Artificial Intelligence (AI) Required, Work From Home The VP of Research, Machine Learning (ML) will own the research and intelligence direction of this system. The VP of Research, Machine Learning (ML) will define how Artificial Intelligence (AI) reasons, evaluates, and improves in a product used with high frequency. This position is 100% Remote. MUST BE WILLING TO TAKE A 60 MINUTE CODING ASSESSMENT. VP of Research, Machine Learning (ML) Responsibilities:

Easy Apply

Full-time

$225,000 - $275,000

Machine Learning Lead Engineer, Machine Learning (ML) Required, Work From Home - G

Remote

•

Today

Machine Learning Lead Engineer, Machine Learning (ML) Required, Work From Home As Machine Learning Lead Engineer, you own the execution layer of our intelligence, turning research and model capabilities into reliable, scalable production systems. You will work across the model lifecycle: data, training, evaluation, inference, and deployment. This is a hands-on leadership role for someone who wants to operate at the intersection of research, systems, and product. Machine Learning (ML) experien

Easy Apply

Full-time

$160,000 - $190,000

Machine Learning Technical Lead, Artificial Intelligence (AI) Required, Work From Home - G

Remote

•

Today

Machine Learning Technical Lead, Artificial Intelligence (AI) Required, Work From Home As Machine Learning Technical Lead, you own the execution layer of intelligence. You will translate research direction into reliable, scalable, production-grade ML systems. This role sits at the intersection of research, infrastructure, and product. You will be responsible for making models trainable, deployable, observable, and performant under real-world constraints. This position is 100% Remote. MUST BE

Easy Apply

Full-time

$190,000 - $225,000

Member of Technical Staff, Machine Learning, Artificial Intelligence (AI) Required, Work From Home - G

Remote

•

Today

Member of Technical Staff, Machine Learning, Artificial Intelligence (AI) Required, Work From Home As a Member of Technical Staff, Machine Learning, you will build core ML components. The Member of Technical Staff will work on real production systems from day one, learning how large-scale ML behaves outside of research settings. The Member of Technical Staff role is for engineers who want to develop strong systems judgment by shipping, debugging, and iterating on real-world ML. This position

Easy Apply

Full-time

$160,000 - $190,000

Principal Machine Learning Engineer, Artificial Intelligence (AI) Required, Work From Home - G

Remote

•

Today

Principal Machine Learning Engineer, Artificial Intelligence (AI) Required, Work From Home As a Principal Machine Learning Engineer, you are a deep technical authority responsible for designing and evolving the most critical ML systems in the company. The Principal Machine Learning Engineer will operate across training, inference, evaluation, and infrastructure, solving the hardest architectural and performance problems. While Technical Leads may own execution at the team level, you set the te

Easy Apply

Full-time

$170,000 - $200,000

Machine Learning (ML) Engineer, Ranking & Retrieval, Machine Learning (ML) and Artificial Intelligence (AI) Required, Work From Home - M

Remote

•

Today

Machine Learning (ML) Engineer, Ranking & Retrieval, Machine Learning (ML) and Artificial Intelligence (AI) Required, Work From Home The Machine Learning (ML) Engineer will build and improve Machine Learning (ML) systems that power our AI-native applications. You will work on classification, retrieval, ranking, personalization and evaluation, taking ownership from model development to production deployment. This role is for engineers who enjoy solving practical Machine Learning (ML) problems,

Easy Apply

Full-time

$120,000 - $160,000

Applied Machine Learning (ML) Engineer, Machine Learning (ML) and Artificial Intelligence (AI) Required, Work From Home - M

Remote

•

Today

Applied Machine Learning (ML) Engineer, Machine Learning (ML) and Artificial Intelligence (AI) Required, Work From Home The Applied Machine Learning (ML) Engineer will build and improve Machine Learning (ML) systems that power our AI-native applications. You will work on classification, retrieval, ranking, personalization and evaluation, taking ownership from model development to production deployment. This role is for engineers who enjoy solving practical Machine Learning (ML) problems, impro

Easy Apply

Full-time

$120,000 - $160,000

LLM Application Engineer, Artificial Intelligence (AI) Required, Work From Home - M

Remote

•

Today

LLM Application Engineer, Artificial Intelligence (AI) Required, Work From Home As an LLM Application Engineer, you will build the intelligence layer that powers the AI experiences. You will work at the intersection of Large Language Models (LLMs), software engineering, and product designing agent workflows, improving model behavior, and turning Artificial Intelligence (AI) capabilities into reliable user experiences. You will own problems end-to-end, from understanding user needs, designing A

Easy Apply

Full-time

$130,000 - $175,000

Machine Learning Platform Engineer, Machine Learning (ML) and Artificial Intelligence (AI) Required, Work From Home - G

Remote

•

Today

Machine Learning Platform Engineer, Machine Learning (ML) and Artificial Intelligence (AI) Required, Work From Home As the Machine Learning Platform Engineer, you will build the infrastructure and systems that power Artificial Intelligence (AI) capabilities. You will design and operate the systems behind the Artificial Intelligence (AI) stack, from model training and evaluation to deployment, inference, observability, and continuous improvement. You will work closely with Artificial Intelligen

Easy Apply

Full-time

$140,000 - $180,000

Software Engineer/Developer (AI Focus) -

Remote

•

8d ago

*This role is 100% Remote! The Software Engineer/Developer AI serves as a senior, hands-on full-stack AI engineer and technical authority, leading the technical strategy, design, and delivery of large-scale mission-critical AI systems supporting federal programs (e.g., HUD, AIR platform). This role combines senior technical leadership, hands-on expertise in Python-based AI/ML systems (including large language models), and ownership of enterprise architecture, governance, and innovation. Serve a

Easy Apply

Full-time

$165,000 - $195,000

Staff Software Engineer, Machine Learning (Consumer Revenue)

Remote or San Francisco, California

•

Today

Discord has a highly engaged community of millions of daily active users who use the platform for many different reasons, but there's one thing that nearly everyone does: play video games. Discord plays a uniquely important role in the future of gaming, and we are focused on making it easier and more fun for people to hang out before, during, and after playing games. We're looking to hire a Staff Machine Learning Engineer on our Consumer Revenue ML team. This team applies Machine Learning acros

Full-time

USD 272,000.00 - 340,000.00 per year

Senior Computer Vision Engineer

Remote or Irvine, California

•

Today

Anduril Industries is a defense technology company with a mission to transform U.S. and allied military capabilities with advanced technology. By bringing the expertise, technology, and business model of the 21st century's most innovative companies to the defense industry, Anduril is changing how military systems are designed, built and sold. Anduril's family of systems is powered by Lattice OS, an AI-powered operating system that turns thousands of data streams into a realtime, 3D command and c

Full-time

USD 220,000.00 - 292,000.00 per year

Staff Machine Learning Engineer

Remote or New York, New York

•

14d ago

Job Description: Integral Ad Science (IAS) is a global technology and data company that builds verification, optimization, and analytics solutions for the advertising industry, and we're looking for a Staff Machine Learning Engineer on the Data Science Team. If you are excited by technology that has the power to handle hundreds of thousands of transactions per second; collect tens of billions of events each day; and evaluate thousands of data points in real-time all while responding in just a f

Full-time

USD 135,100.00 - 231,600.00 per year

Senior/Staff Engineer, Machine Learning - Online Mapping

Mountain View, California

•

Today

Who We Are Nuro believes self-driving vehicles are the most immediate and profound opportunity for AI to drive positive change in the physical world. Safer streets, more time for what matters, and easier access to the world around us, that's why we're building a universal autonomy platform: self-driving for all roads and all rides. Founded in 2016, Nuro is a physical AI company developing Level 4 autonomous driving technology for a wide range of vehicles, use cases, and markets. Powered by the

Full-time

USD 193,930.00 - 291,150.00 per year

JAX jobs in California, USA