Machine Learning Jobs in Santa Clara, CA

Lead design, training, and deployment of large-scale ML models (LLMs, diffusion, GNNs). Own end-to-end training pipelines, distributed training, evaluation, and inference architecture. Collaborate with data and infra teams, drive reproducible model development, and provide technical leadership and mentorship from research through production.

As a Staff Machine Learning Engineer, you will lead the development and deployment of AI models for autonomous driving, focusing on safety and reliability. Responsibilities include driving the roadmap, training models, and collaborating across teams.

The role involves deploying and optimizing LLMs, designing the ML serving stack, and ensuring high-performance GPU services for production readiness.

Lead junior team members in test planning, automation, and execution of software for AI compute engine, developing test suites and infrastructures, and collaborating on test coverage and results.

The AI/ML Engineer designs and operates a production-grade orchestration platform, focusing on multi-agent workflows, LLM integration, and backend systems. Responsibilities include managing cloud-native infrastructure, ensuring system reliability and observability.

As a Machine Learning Engineer Intern, you'll bridge the sim-to-real gap in data for autonomous truck software, optimizing and integrating BEV features and supporting self-play reinforcement learning.

Join PlusAI as a Machine Learning Engineer Intern to develop AI-based features for autonomous trucks, focusing on multi-modal data processing and event classification.

The intern will prototype a generative simulator for diverse driving scenarios, evaluate existing tools, and integrate outputs into data pipelines for autonomous driving applications.

The Technical Product Manager will lead product direction, interface with engineering teams, and ensure the successful deployment of AI-powered safety systems. Responsibilities include defining product vision, translating customer needs, and maintaining high product quality.

Lead the productization of AlpaSim and AlpaDreams technologies. Oversee architecture integration, manage team performance, and transform prototypes into reliable tools, focusing on developer experience and production excellence.

Build and deploy production AI/LLM systems and agentic workflows for engineering productivity, diagnostics, search, and automation. Design retrieval, tool integration, evals, failure handling, and reusable skills; collaborate with infrastructure and domain teams to monitor and improve systems in production.

Build and operate ML infrastructure for LLMs and agent systems: model gateways, routing, serving, telemetry, observability, evaluation, cost-aware routing, dashboards, and APIs/SDKs to enable reliable production AI at scale.

Design and implement agent harnesses and post-training pipelines, build evaluation benchmarks, develop RL environments and reward models, and debug/optimize multi-GPU training runs to improve LLM capabilities for agentic applications.

The Senior Machine Learning Engineer will develop and optimize AI image generation models, monitor production issues, and collaborate on ML solutions for generative features.

Design and implement ML solutions for optimizing GPU accelerated Apache Spark workloads. Collaborate on deployment and lead technical teams in ML.

The Applied Machine Learning Research Scientist will translate ML techniques into scalable systems, improve LLM performance, and optimize workflows while collaborating closely with researchers and engineers.

The role involves developing large-scale data processing pipelines, automating data selection, and collaborating on ML model improvements to enhance vehicle performance and safety.

Engineer will leverage AI for pre-silicon verification, collaborate with teams, develop AI solutions, and engage customers regarding verification strategies.

Design, develop, and implement machine learning models and collaborate with teams to enhance services with AI/ML solutions.

Contribute to developing machine learning models for multi-scale biology, focusing on design, deployment and optimization across teams and technologies.

The Senior Machine Learning Engineer will design, deploy, and operate ML systems for advertising, collaborating with cross-functional teams to enhance performance and reliability. Responsibilities include mentoring, promoting best practices, and owning ML project lifecycles.

Seeking a PhD research intern to develop algorithms for robot manipulation, working on AI models and collaborating in a robotics startup environment.

Build and automate large-scale ML data pipelines and tooling for autonomous trucking. Improve dataset curation, deduplication, labeling, evaluation, and metrics infrastructure. Collaborate with robotics, autonomy, and infra teams to support continuous learning, deployment, and scalable AI infrastructure for Kodiak's fleet.

The Principal Compiler Engineer will lead compiler engineering, optimizing ML model performance and coordinating with development teams to innovate compiler infrastructure.

The ML Features Solutions Engineer will develop, optimize, and deploy core machine learning features for enterprise applications while collaborating with cross-functional teams to ensure production-grade quality and performance.