Machine Learning Jobs in Santa Clara, CA

As a Senior ML Infrastructure Engineer, you will design scalable architectures for ML, build robust data pipelines, manage GPU clusters, and drive improvements in CI/CD workflows while collaborating with cross-functional teams.

As a Senior Machine Learning Engineer, you will own the ML stack, develop generation systems, improve quality, collaborate cross-functionally, and build data pipelines.

The Machine Learning Engineer will build and scale ML infrastructure, drive model performance, and ensure reliability while collaborating with various teams to optimize feature engineering pipelines and champion best practices.

Lead the design and implementation of GPU compute clusters for deep learning and high-performance workloads, manage large-scale systems and ensure efficient resource utilization.

Optimize machine learning models for embedded environments, focusing on performance, efficiency, and deployment strategies across various platforms.

The Machine Learning Engineer role at Wayve focuses on leading initiatives in model-based autonomous driving, involving the development and delivery of ML-driven behaviors, collaboration across teams, and mentoring other engineers.

The intern will profile, optimize, and automate training processes for BEV models, developing custom kernels and leveraging LLMs for performance improvements.

As a Senior MLOps Engineer, you will optimize machine learning models for performance and scalability, collaborate with R&D engineers, and implement CI/CD pipelines.

As an ML Engineer specializing in inference, you will optimize models for production, implement neural network techniques, and collaborate with researchers.

As an ML Engineer at Luma AI, you will design high-performance model serving pipelines, manage GPU resources, and optimize CI/CD for large-scale AI systems.

Lead the navigation ML workstream for autonomous driving systems, overseeing model training and deployment, defining technical direction, and mentoring junior engineers.

Work on vision-language model development and infrastructure: improve VLM understanding, OCR and function-calling workflows, fine-tuning recipes, evaluation and robustness, and optimize training/serving pipelines. Own end-to-end projects from experiments to production-ready code, collaborating on scale, cost, and reliability.

Build and scale the infrastructure layer for visual intelligence: optimize VLM inference and GPU serving, design multimodal APIs and structured outputs, ensure scalable backend systems, reliability, observability, CI/CD, and developer experience. Drive 0→1 product work with strong testing discipline, schema validation, and performance optimization.

Lead RTL development and microarchitecture for an AI-enabled RISC-V CPU core, including vector and matrix engines. Define features, optimize performance/power/area, support verification and simulation, collaborate with CPU modeling and physical design teams, and ensure design meets timing, reliability, testability, and power goals.

The role involves designing scalable data pipelines for 3D, video, and sensor data, optimizing infrastructure, and productionizing ML models with researchers.

The role involves collaborating with customers to solve engineering challenges using machine learning, develop solutions, train teams, and improve products based on feedback.

Lead the development of AI/ML deployment infrastructure, collaborate with engineers, develop pipelines, manage teams, and enhance data applications.

Develop AI/ML models for 3D simulation and geology prediction, refine data pipelines, and integrate deep learning tech with product backend.

Responsible for software integration and quality for Cerebras AI platform, focusing on automation, debugging, and cross-team collaboration. Develops QA strategies and testing methodologies to ensure product quality.

Lead design, build, deploy, and maintain production-grade AI/ML systems (LLMs, agentic architectures, classical ML). Provide hands-on technical leadership across architecture, model selection, implementation, testing, deployment, observability, and operational support while collaborating with data scientists, platform and security teams and mentoring engineers.

Lead development of large-scale ranking, personalization, and recommendation systems for Roku's entertainment assistant. Design and deploy ML models (LLMs, transformers, RL, bandits), build feature platforms and data pipelines, run online A/B experiments, and mentor engineers while balancing engagement, retention, relevance, and monetization objectives.

Research and build efficient ML systems for large-scale LLMs and agentic RL: design algorithms and system techniques, prototype in training/inference stacks, run large-scale experiments, and translate findings into production or publications.

Design and implement AI-driven vehicle intelligence features using telemetry, user behavior, navigation, and fleet data. Apply reinforcement learning, forecasting, planning, personalization, and simulation to optimize energy, comfort, charging, and routing. Validate in simulation and integrate models into vehicle and cloud platforms focusing on latency, efficiency, safety, explainability, and OTA readiness while collaborating with autonomous driving, product, UX, and systems teams.

Design and develop ML/AI vehicle intelligence features using telemetry, navigation, user behavior, and fleet data. Build predictive, optimization, personalization, and RL-based models; validate via simulation and digital twins; integrate models into vehicle and cloud platforms with attention to latency, compute efficiency, safety, and OTA readiness. Collaborate cross-functionally with product, UX, systems, controls, and autonomous driving teams.

Lead workload-driven AI/ML architecture across hardware and software: characterize workloads, perform system-level performance and memory analysis, define measurement and pre-silicon estimation methods, leverage MLIR/IREE toolchains, identify bottlenecks, recommend SoC/CPU/memory optimizations, and present findings to engineering leadership while mentoring junior engineers.