Machine Learning Jobs in San Jose, CA

Develop and validate NVIDIA PhysicsNemo, a Physics-AI framework for digital twins and simulation surrogates. Collaborate with internal teams and external partners, implement state-of-the-art deep learning architectures for scientific/engineering simulations, and stay current with research to improve production-ready ML and HPC solutions.

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

As a founding Data Scientist/Machine Learning Engineer, you'll develop AI/ML models, enhance product capability, and drive impactful user outcomes while working closely with product teams.

As a Senior Software Architect for AI Networking, you'll design and implement AI networking technologies, optimize data processing in AI systems, and collaborate on cutting-edge solutions.

The intern will design and implement AI-driven automation workflows to enhance the silicon testing process, collaborate with engineers, and document automation recipes.

Assist in developing software for autonomous driving vehicles, participating in project work involving machine learning, coding, and testing.

The Product Development Engineer drives testability and manufacturability of circuits, performs yield analysis using data science techniques, and interfaces with various engineering teams to optimize silicon yield and quality.

The engineer will leverage AI to enhance pre-silicon verification processes, collaborate with teams, and improve methodologies using advanced tools.

Develop and manage a Machine Learning platform, implementing user interfaces, ensuring security, optimizing performance, and collaborating with data scientists.

Design, train, and optimize ML models for LiDAR perception; develop end-to-end ML workflows (data pipelines, training, metrics); productize models into NVIDIA DRIVE AV with efficient C++ production code; collaborate across teams and incorporate state-of-the-art ML techniques for real-time autonomous driving perception.

Design and build a scalable failure-attribution framework for heterogeneous CPU/GPU clusters: collect high-fidelity telemetry, implement low-overhead tracing, automate root-cause classification (hardware, software, environment), and enable proactive resiliency measures like migration or checkpointing.

Lead and coordinate cross-functional engineering programs for deep learning software and GenAI model initiatives. Plan, forecast, track dependencies, report status, implement metrics, guide agile practices, moderate technical discussions, and drive organizational alignment to meet business objectives.

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 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 collaborating with customers to solve engineering challenges using machine learning, develop solutions, train teams, and improve products based on feedback.

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.

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

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.

The role involves developing ML-based planning algorithms for autonomous trucks, ensuring safety checks, real-time operation, and keeping up with research. Requires strong communication and collaboration.

Develop and deploy end-to-end AI/ML solutions focused on GenAI and LLMs. Build and fine-tune models, create embedding search and agentic systems, perform forecasting and data mining, collaborate with stakeholders, and drive improvements in supply chain and material demand forecasting.

As a Senior Staff Software Engineer, you will develop and maintain AI deployment software, optimizing hardware-software co-design, and work closely with various experts to enhance infrastructure.

Lead research, design, and implementation of classical and ML/DL computer vision algorithms for defect detection, classification, metrology, and image analysis on semiconductor inspection tools. Prototype and evaluate approaches, run feasibility studies and benchmarking on large datasets, optimize performance for available compute (GPU), write production-quality code, and collaborate with software, systems, and compute infrastructure teams to productize solutions.

Define and execute the scientific roadmap for search and recommendation: architect large-scale ranking and retrieval systems, develop and distill Ranking Foundation Models, integrate vector/RAG/hybrid retrieval into production, partner with product and engineering, and lead and scale multidisciplinary ML teams to optimize marketplace health.