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

•

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

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

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

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

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

Software Engineer/Developer (AI Focus) -

Remote

•

2d 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

#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

•

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

Software Development Engineer, AI/ML, AWS Neuron, Model Inference

Cupertino, California

•

Today

The Annapurna Labs team at Amazon Web Services (AWS) builds AWS Neuron, the software development kit used to accelerate deep learning and GenAI workloads on Amazon's custom machine learning accelerators, Inferentia and Trainium. The AWS Neuron SDK, developed by the Annapurna Labs team at AWS, is the backbone for accelerating deep learning and GenAI workloads on Amazon\'s Inferentia and Trainium ML accelerators. This comprehensive toolkit includes an ML compiler, runtime, and application framewor

Full-time

USD 165,200.00 - 223,600.00 per year

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

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

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

Remote

•

Today

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

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

Lead AI Engineer

Santa Clara, California

•

17d ago

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 systems end to end across multiple domains: Generative and Agentic AI (including working with foundation models such as Claude), Computer Vision on u

Easy Apply

Contract

Depends on Experience

Fulltime | Senior Java Developer | 100% Remote

Remote

•

28d ago

Job Title: Senior Java Developer Work Setting: 100% Remote Job Type: Fulltime (Permanent) NO SPONSORSHIP AVAILABLE JOB DETAILS: 10+years in IT with strong, hands-on Java background Core Java, Spring / Spring Boot, Microservices architecture REST API design and integration patterns AWS/Azure/Google Cloud Platform Cloud Experience Solid understanding of application security, performance tuning, and scalability for enterprise systems Strong JavaScript/TypeScript development experience

Easy Apply

Full-time

$80000 - $120000

JAX jobs in California, USA