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

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

Remote

•

Yesterday

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

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

Remote

•

Yesterday

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

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

Remote

•

Yesterday

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 - $190,000

Sr. Software Engineer (Vehicle Engineering)

Hawthorne, California

•

Today

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. SR. SOFTWARE ENGINEER (VEHICLE ENGINEERING) Be a member of the Artificial Intelligence Software Engineering team focusing on solving complex AI data problems for our launch vehicles and spacecrafts. Our team is creat

Full-time

USD 160,000.00 - 225,000.00 per year

ML Engineer, Surrogate Modeling (Vehicle Engineering)

Hawthorne, California

•

Today

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. ML ENGINEER, SURROGATE MODELING (VEHICLE ENGINEERING) Be a member of the AI for Vehicle Engineering team, focusing on developing high-performance surrogate models to solve complex physics and engineering problems for

Full-time

USD 125,000.00 - 145,000.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

#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

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

Remote

•

Yesterday

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) experience

Easy Apply

Full-time

$160,000 - $190,000

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

Remote

•

Yesterday

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

Java Software Engineer - Senior-Level

Remote or Almont, Colorado

•

Today

Description: Remote Our client seeks a senior-level Java Software Engineer to develop and maintain scalable Java applications within an automated DevSecOps AWS cloud environment supporting the Department of Commerce. The role includes designing cloud-native solutions, building CI/CD pipelines with integrated testing and security, implementing database and storage solutions, and ensuring performance, resiliency, and security across applications and infrastructure. The team operates in Agile, wi

Contract

Java Software Engineer - Mid-Level

Remote or Almont, Colorado

•

Today

Description: Remote Our client seeks a mid-level Java Software Engineer to develop and maintain scalable Java applications in an automated DevSecOps AWS environment supporting the Department of Commerce. You will build cloud-native services, implement CI/CD pipelines with integrated testing and security scans, and ensure solutions meet reliability, performance, and security standards. You will collaborate within Agile teams, support infrastructure automation, and monitor system health with eff

Contract

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

Remote

•

Yesterday

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

Easy Apply

Full-time

$150,000 - $170,000

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

Remote

•

Yesterday

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 Intelligenc

Easy Apply

Full-time

$140,000 - $180,000

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

Remote

•

Yesterday

Senior Machine Learning Engineer, Artificial Intelligence (AI) Required, Work From Home As the Senior Machine Learning Engineer, you are an independent owner of critical Machine Learning (ML) subsystems in production. You take ambiguous problems, design practical solutions, and ship systems that operate reliably at scale. This is a hands-on, high-impact role focused on depth. This position is 100% Remote. MUST BE WILLING TO TAKE A 60 MINUTE CODING ASSESSMENT. Senior Machine Learning Engineer

Easy Apply

Full-time

$170,000 - $190,000

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

Remote

•

Yesterday

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

Data Scientist

Santa Clara, California

•

11d ago

Title: Lead AI Engineer / Data Scientist Focus area: GenAI, Agentic & Computer Vision Solutions Location: On-site (Santa Clara) About the Role We are looking for a Lead AI Engineer / Data Scientist: an AI-engineering-heavy practitioner who combines deep algorithmic and deep learning expertise with strong solutioning and customer-facing skills. This is a hybrid role: part senior AI engineer, part technical lead, and part trusted advisor to customers. You will design and build production AI sy

Easy Apply

Contract, Third Party

$70 - $80

Lead Data Scientist

Santa Clara, California

•

13d ago

Key Responsibilities Solution Design & Technical Leadership: Design end-to-end ML/LLM and agentic solutions, from problem framing and data strategy through to deployment and monitoring. Architect agentic systems: multi-step, tool-using, and multi-agent workflows: including orchestration, tool/function integration, memory, and guardrails. Own the technical architecture for solutions: model selection, fine-tuning approach, agent/orchestration design, serving strategy, API design, and infrastruct

Easy Apply

Third Party, Contract

$80 - $100

JAX jobs in California, USA