247 ml framework engineer jobs at 120 companies in San Pablo, 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
5d
Save
Mark Applied
Hide
AI/ML Engineer
San Francisco, California, United States
$150k-$250k/yr OnsiteFull Time
Clera
Clera: AI talent agent matching professionals with high-growth startup roles
4+ YOE4+ years developing and deploying AI/ML models, 2+ years technical leadership, proficiency in Python and modern ML frameworks, experience with physics-informed/spatial data and MLOps; BS required.
Python, PyTorch, JAX, TensorFlow, C++, Java, ANSYS, SolidWorks, Abaqus, COMSOL, CREO, Autodesk Inventor
1mo
Save
Mark Applied
Hide
Sr. ML Engineer – ML & Applied AI
San Francisco, California, United States
$181k-$236k/yr OnsiteFull Time
Gap Inc.
Gap Inc.NYSE: GAP: Global specialty retailer of apparel and accessories.
10+ YOE10+ years building production ML systems; strong Python and software engineering; experience with ML frameworks, model serving, MLOps, cloud platforms, and distributed data processing.
Python, scikit-learn, XGBoost, PyTorch, TensorFlow, FastAPI, Flask, Docker, Kubernetes, GCP, AWS, Azure, Git, Spark, PySpark, Databricks
1mo
Save
Mark Applied
Hide
ML Systems & Performance Engineer
San Francisco, California, United States
OnsiteFull Time
Engram
Engram: Developing persistent memory layers for enterprise AI systems.
5+ YOE5+ years building training/inference systems; strong engineering skills; experience with ML frameworks, GPUs, distributed systems; bachelor's degree or equivalent experience.
PyTorch, JAX, GPUs
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
13h
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
1mo
Save
Mark Applied
Hide
Staff Software Engineer - AI/ML
New York City or San Francisco
$192k-$260k/yr OnsiteFull Time
Databricks
Databricks: A unified platform for data analytics and artificial intelligence.
10+ YOE10+ years engineering experience building and shipping ML/LLM products; hands-on in Python and modern ML frameworks (e.g., PyTorch); experience with personalization, model evaluation/monitoring, RAG, fine-tuning, and translating customer problems into product solutions.
Python, PyTorch, Apache Spark, Delta Lake, MLflow
3mo
Save
Mark Applied
Hide
Sr. ML Engineer – ML & Applied AI
San Francisco, California, United States
OnsiteFull Time
Gap
GapNYSE: GAP: Retailer of apparel and accessories across multiple global brands.
10+ YOE10+ years in machine learning or software engineering; production ML systems; Python; ML frameworks; CI/CD; cloud; Spark; Databricks; Docker; Kubernetes.
Python, scikit-learn, XGBoost, PyTorch, TensorFlow, FastAPI, Flask, Docker, Kubernetes, GCP, AWS, Azure, Spark, PySpark, Databricks, Git, SQL
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 Vehicle Motion Control Engineer – Vehicle System Controls
Milford or Mountain View
$218k-$275451k/yr HybridFull Time
General Motors
General MotorsNYSE: GM: Manufactures and sells automobiles and automotive parts globally.
8+ YOEMSc/PhD in controls-related field; 8+ years in control systems and embedded software for vehicle motion; experience with C/C++, MATLAB/Simulink, vehicle dynamics tools (CarSim), Vehicle SPY/INCA/CANalyzer, Python and ML frameworks; ISO 26262/SOTIF familiarity.
Python, PyTorch, TensorFlow, scikit-learn, NumPy, pandas, MATLAB/Simulink, C, C++, CarSim, Vehicle SPY, INCA, CANalyzer, MIL, SIL, HIL, DiL
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