183 ml framework engineer jobs at 73 companies in Watsonville, CA

3w
Save
Mark Applied
Hide
ML Framework (MetalLM) Engineer
Cupertino, California, United States
OnsiteFull Time
Apple
AppleNASDAQ: AAPL: Designs and sells consumer electronics, software, and online services.
Systems engineering experience building scalable, efficient production-grade solutions for high-throughput GPU execution and distributed inference; familiarity with PyTorch, JAX, and Metal.
PyTorch, JAX, Metal
1w
Save
Mark Applied
Hide
ML Systems Research Engineer, RL / Inference / Agent Systems
Santa Clara, California, United States
HybridFull Time
AMD
AMDNASDAQ: AMD: Designs and manufactures computer processors and graphics technology.
Experienced ML systems engineer with strong Python and ML framework skills, experience in RL/inference systems, distributed experimentation, and GPU/infrastructure workflows; advanced degree preferred.
Python, PyTorch, JAX, TensorFlow, Kubernetes, Ray, Slurm, ROCm, HIP, CUDA
1mo
Save
Mark Applied
Hide
Matterport - Senior ML Ops Engineer
Sunnyvale, California, United States
$173k-$253k/yr HybridFull Time
CoStar Group
CoStar GroupNASDAQ: CSGP: Global provider of commercial and residential real estate data.
3+ YOEBachelor's or equivalent experience, 3+ years ML engineering focused on model optimization/deployment, proficiency in Python, ML frameworks (TensorFlow/PyTorch), cloud (AWS/Azure/GCP), Git, and strong communication.
Python, TensorFlow, PyTorch, AWS, Azure, GCP, Git, Temporal, Airflow, Kubeflow, Docker, Kubernetes, CI/CD, GPUs, CPUs
1d
Save
Mark Applied
Hide
Lead AI / ML Engineer
New York City or Milwaukee or Dallas or Columbus or Kirkland or Cincinnati or Cleveland or Oklahoma City or Austin or Albany or Chicago or St. Petersburg or Hartford or Pittsburgh or St. Louis or Miami or Sacramento or Raleigh or Minneapolis or Mountain View or Scottsdale or San Francisco or Morristown or Denver or Boston or Philadelphia or Des Moines or Overland Park or Los Angeles or Charlotte or Walnut Creek or Carmel or Seattle or Houston or Arlington or Atlanta or Redmond or Bentonville or Beaverton or Nashville or Detroit or San Diego
$80k-$294k/yr HybridFull Time
Accenture
AccentureNYSE: ACN: Global provider of management consulting and technology services.
6+ YOEBachelor's degree or equivalent experience, 6+ years developing and deploying AI/ML solutions, programming proficiency, experience with ML frameworks, big data, databases, and cloud platforms.
Python, R, Java, TensorFlow, PyTorch, Hadoop, Spark, AWS, Microsoft Azure, Google Cloud, D3.js, ggplot
20h
Save
Mark Applied
Hide
ML Engineer, Foundation Model Infrastructure
Mountain View or San Francisco or Kirkland or New York City
$175k-$215k/yr HybridFull Time
Waymo
Waymo: Autonomous driving technology for ride-hailing and logistics.
Master's degree or equivalent practical experience; Python and C++ proficiency; modern deep learning framework familiarity; experience with large-scale data pipelines or ML infrastructure.
Python, C++, Flume, JAX, PyTorch, TensorFlow, Spark, Borg, Kubeflow
3w
Save
Mark Applied
Hide
Senior Principal Applied ML Engineer
San Jose, California, United States
$154k-$286k/yr OnsiteFull Time
Cadence Design Systems
Cadence Design SystemsNASDAQ: CDNS: Develops computational software and hardware for electronic system design.
5+ YOEAdvanced ML and software engineering experience building production-quality agentic systems, LLM deployment, evaluation frameworks, retrieval pipelines, and observability for large-scale EDA workflows.
Xcelium, Jasper, Palladium, Protium, Helium, EDA, Large language models (LLMs), Retrieval-augmented generation (RAG)
1d
Save
Mark Applied
Hide
Principal AI/ML Research Engineer
United States or California or San Jose or Seattle or Portland or Boston or Chicago
$250k-$289k/yr RemoteFull Time
WEX
WEXNYSE: WEX: Provides global payment processing and business information management services.
12+ YOE12+ years in software or ML engineering, including 5+ years in applied AI/ML research. Requires advanced expertise in model architecture, algorithms, Python, deep learning frameworks, distributed AI, cloud, and ML platforms.
Python, PyTorch, TensorFlow, JAX, C++, Java, Ray, DeepSpeed, Megatron, Spark, AWS, Azure, SageMaker, MLflow, Databricks, LanceDB, Pinecone, Qdrant, Milvus, Transformer, Tabular Transformers, LLMs, RAG, LoRA, PEFT, Diffusion, Reinforcement Learning, RLHF, NeurIPS, ICML, KDD, ACL
1mo
Save
Mark Applied
Hide
Staff Agentic ML Engineer - Photoshop
San Jose or Waltham or San Francisco or Seattle or Los Angeles or New York City or California
$190k-$346k/yr OnsiteFull Time
Adobe
AdobeNASDAQ: ADBE: Provides software for digital media creation and marketing analytics
8+ YOEMaster's or PhD in CS/ML/Data Science, 8+ years ML experience, expertise in LLMs, fine-tuning (SFT, RLHF, DPO, PEFT/LoRA), agent system development, evaluation frameworks; proficiency with PyTorch and agentic frameworks; strong communication and problem-solving.
PyTorch, LangChain, LangGraph, MCP, A2A, Agent Development Kit (ADK), Claude Code, Codex, Cursor
5d
Save
Mark Applied
Hide
Founding Engineer – ML Research
Mountain View, California, United States
$220k-$300k/yr OnsiteFull Time
Clera
Clera: AI talent agent matching professionals with high-growth startup roles
3+ YOE3+ years ML research/applied ML experience; proficiency in Python and ML frameworks (PyTorch,JAX,TensorFlow); experience with Transformers, diffusion models, RLHF; building scalable experimentation pipelines; strong documentation and reproducibility skills.
Python, PyTorch, JAX, TensorFlow
1d
Save
Mark Applied
Hide
Staff ML Infra Engineer
Mountain View, California, United States
$174k-$299k/yr OnsiteFull Time
Coupang
CoupangNYSE: CPNG: Provides an end-to-end e-commerce and logistics network.
5+ YOEBachelor's in CS/EE/math,5+ years applied ML experience,proficiency in Python/Java,experience with big data pipelines,ML frameworks,cloud platforms,and building scalable low-latency services.
Python, Java, Hadoop, Hive, Presto, Spark, Scala, Apache Airflow, TensorFlow, PyTorch, AWS, GCP
1mo
Save
Mark Applied
Hide
Senior Software Engineer, ML Infrastructure
Mountain View, California, United States
$194k-$291k/yr OnsiteFull Time
Nuro
Nuro: Builds autonomous driving software and electric delivery robots.
4+ YOE4+ years in ML infrastructure or distributed systems; experience with IaC (Terraform/Pulumi/Crossplane), workload orchestrators (Kubernetes, Ray, Slurm), distributed data frameworks (Apache Spark/Beam), and feature stores (Feast/Hopsworks/Redis).
Terraform, Pulumi, Crossplane, Kubernetes, KubeRay, Ray, Slurm, Volcano, Apache Spark, Apache Beam, Feast, Hopsworks, Redis, Lustre, Ceph, NVMe, AWS, GCP, Azure, CNCF, Kubeflow
1mo
Save
Mark Applied
Hide
AI/ML Engineering Manager
Milpitas, California, United States
$170k-$210k/yr OnsiteFull Time
Corsair
CorsairNASDAQ: CRSR: Sells high-performance gaming peripherals and PC hardware components.
6+ YOE1+ Mgmt6+ years in AI/ML engineering with 1+ years technical lead experience; strong Python and SQL; experience with LLM frameworks, workflow orchestration, data engineering, and MLOps; ability to architect agentic AI systems and lead engineers.
Python, SQL, LangChain, LangGraph, CrewAI, Strands, AWS Bedrock, Azure AI Foundry, Microsoft Fabric, Temporal, Airflow, Step Functions, dbt, Dagster, Spark, MLflow, W&B, Docker, Kubernetes, CI/CD, LoRA, PEFT
1mo
Save
Mark Applied
Hide
Staff ML Infra Engineer, Search & Discovery
Mountain View, California, United States
$174k-$299k/yr OnsiteFull Time
Coupang
CoupangNYSE: CPNG: Provides online retail, grocery delivery, and video streaming services.
5+ YOEBachelor's in CS/EE/math/stats, 5+ years applied ML experience, production ML systems experience, proficiency in Python/Java, experience with big data, ML frameworks, and cloud platforms.
Python, Java, Hadoop, Hive, Presto, Spark, Scala, Apache Airflow, TensorFlow, PyTorch, AWS, GCP
1mo
Save
Mark Applied
Hide
Staff AI/ML Architect, Embodied AI
Sunnyvale, California, United States
$230k-$330k/yr OnsiteFull Time
Intuitive
IntuitiveNASDAQ: ISRG: Robotic-assisted systems for minimally invasive surgery.
10+ YOEMS or PhD plus 10+ years building AI/ML models and end-to-end system architectures for autonomous or embodied systems; expertise in hierarchical perception/reasoning/control, real-time safety-critical ML, and hands-on experience with modern deep learning frameworks and C++/Python.
PyTorch, TensorFlow, JAX, Python, C++, JEPA, MAE, DINO, NVIDIA Isaac Sim, Triton, CUDA, LLM
1mo
Save
Mark Applied
Hide
Lead AI Engineer -- Advanced AI (applied ML, LLMs, agentic AI, ML Ops)
Brooklyn Park or Sunnyvale
$132k-$286k/yr HybridFull Time
Target
TargetNYSE: TGT: Operates a chain of general merchandise stores and supermarkets.
5+ YOEBachelor's in a quantitative field (MS preferred), 5+ years applied ML/AI engineering, hands-on Python and ML framework experience, LLM and agentic systems experience, production ML engineering and system design skills.
Python, PyTorch, TensorFlow, LangChain, LlamaIndex, Semantic Kernel, CI/CD
2w
Save
Mark Applied
Hide
AI/ML Machine Learning Engineer
Herndon or King of Prussia or Littleton or Sunnyvale
$89k-$178k/yr OnsiteFull Time
Lockheed Martin
Lockheed MartinNYSE: LMT: Designs and manufactures global security and aerospace systems.
Bachelor's in STEM, experience with ML/AI and LLM/agentic technologies, strong Python skills, experience with modern ML frameworks, cloud and MLOps, agile development and strong communication.
Python, TensorFlow, PyTorch, Keras, Scikit-learn, AWS, Kubernetes, MLOps, DevOps, CI/CD, Scrum, Kanban, LLM
1mo
Save
Mark Applied
Hide
Staff, Software Engineer - Backend / ML
Sunnyvale, California, United States
$143k-$286k/yr HybridFull Time
Walmart
WalmartNYSE: WMT: Multinational retail operating discount stores and supermarkets.
6+ YOE10+ years building large-scale distributed backend microservices, data pipelines, and ML-serving infrastructure; expertise with Java/Spring Boot, Kubernetes, Spark, Kafka, NoSQL, testing frameworks, and cloud data platforms; experience leading engineering teams.
Java, Spring Boot, Kubernetes, Apache Spark, SparkSQL, Kafka, Cassandra, JUnit, Mockito, Hive, BigQuery, GCP, GCS, Cosmos DB, Azure Blob Storage, Solr, Elasticsearch, Vespa, FAISS, PyTorch, TensorFlow, scikit-learn, XGBoost, TensorRT, ONNX, Python, Airflow
3w
Save
Mark Applied
Hide
Senior Software Engineer, ML Compilers, TPU, YouTube
San Bruno or Mountain View
$174k-$253k/yr OnsiteFull Time
Google
GoogleNASDAQ: GOOGL: Provides online search, advertising, cloud computing, and consumer electronics.
8+ YOEBachelor's degree or equivalent, 8+ years C++/Python programming, 5+ years launching software, 3+ years software design, experience with ML compilers, ML frameworks, and compiler optimization.
C++, Python, TensorFlow, JAX, PyTorch, Accelerated Linear Algebra (XLA), Pallas, TPU
1w
Save
Mark Applied
Hide
Sr AI Engineer - Advanced AI (ML Ops, LLMs, Agentic workflows)
Brooklyn Park or Sunnyvale
$98k-$211k/yr HybridFull Time
Target
TargetNYSE: TGT: General merchandise retailer operating physical stores and e-commerce.
5+ YOE4-year degree or equivalent,5+ years software engineering experience building production AI/ML systems,proficiency in Python and ML frameworks,experience with model APIs,cloud,CI/CD,observability and production support.
Python, PyTorch, TensorFlow, LangChain, LlamaIndex, Semantic Kernel
4w
Save
Mark Applied
Hide
Staff Research Engineer, LLM - TikTok Ads Core ML, Ranking
San Jose, California, United States
$245k-$450k/yr OnsiteFull Time
TikTok
TikTok: Global short-form video hosting and social media platform.
MS or above in a quantitative field, strong ML/DL and programming skills, experience with big-data systems and ML frameworks, familiarity with online experimentation and advertising business value.
C/C++, Python, Linux, Hadoop, Spark, TensorFlow, PyTorch, MXNet