67 ml framework engineer jobs at 19 companies in Pacific Grove, CA
3w
Save
Mark Applied
Hide
3w
ML Framework (MetalLM) Engineer
Cupertino, California, United States
OnsiteFull Time
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.
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)
United States or California or San Jose or Seattle or Portland or Boston or Chicago
$250k-$289k/yrRemoteFull Time
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.
San Jose or Waltham or San Francisco or Seattle or Los Angeles or New York City or California
$190k-$346k/yrOnsiteFull Time
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
Staff Research Engineer, LLM - TikTok Ads Core ML, Ranking
San Jose, California, United States
$245k-$450k/yrOnsiteFull Time
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.
OKX: Global cryptocurrency exchange and Web3 digital wallet developer.
Significant ML engineering experience producing production models and LLM agents; strong Python and ML framework experience; model explainability and reliability; experience with risk or high-stakes domains preferred.
Fellow Software Engineer — AI Performance & Reliability
San Jose or Bellevue
$235k-$402k/yrHybridFull Time
AMDNASDAQ: AMD: Designs and manufactures computer processors and graphics technology.
PhD or equivalent in AI/ML/CS, strong software engineering, experience profiling and optimizing ML models and AI workloads, proficiency in Python/C++, ML frameworks, customer-facing troubleshooting and performance analysis.
AmazonNASDAQ: AMZN: Global online retail and cloud computing technology provider.
8+ YOE8+ years technical experience, deep ML model deployment experience, hands-on with deep learning frameworks, customer-facing consulting and architecture skills.
AWS Neuron Software Development Kit (SDK), PyTorch, TensorFlow, JAX
San Francisco or McLean or Cambridge or San Jose or New York City
$294k-$335k/yrOnsiteFull Time
Capital OneNYSE: COF: Financial services offering credit cards, banking, and loans.
6+ YOEBachelor's or Master's in CS/Engineering/AI with 6–8+ years building and deploying scalable AI/ML systems; strong programming in Python/Go/Scala/Java and experience with agentic frameworks, cloud platforms, and LLMs.
RokuNASDAQ: ROKU: Operates a TV streaming platform and sells streaming hardware.
5+ YOE5+ years applied machine learning experience; strong statistical, optimization, and experimental methodology knowledge; proficient in Spark, Python or Java; experience with real-time, low-latency systems and distributed ML frameworks.
ArcherNYSE: ACHR: Develops electric vertical takeoff and landing aircraft for urban mobility.
MS or PhD in CS or Computer Engineering with strong emphasis on AI/ML; strong ML frameworks; Transformer and multimodal models expertise; production-ready research experience.
Altera: Manufacturer of field-programmable gate arrays and programmable logic devices.
10+ YOEBachelor's degree, 10+ years ML engineering/MLOps experience, strong Python, cloud ML platforms (AWS/GCP/Azure), Docker/Kubernetes, CI/CD, ML frameworks (PyTorch/TensorFlow/JAX), experience with MLflow/W&B and HPC schedulers.
Prodapt: Provides IT consulting and network transformation services for telecommunications providers.
8+ YOE8+ years development experience; proficiency in Python and NodeJS, full-stack frameworks, Postgres/NoSQL, and AI/ML frameworks; bachelor’s degree or equivalent.
PayPalNASDAQ: PYPL: Digital platform for sending money and processing online payments.
5+ YOE5+ years experience with ML frameworks; cloud platforms (AWS/Azure/GCP); Bachelor’s degree or equivalent in a technical field; strong ML model development and deployment skills.
ByteDance: Developing AI-driven content platforms and mobile applications.
3+ YOE3+ years developing and deploying large-scale systems; contributions to open-source ML frameworks; strong background in HPC, hardware-software co-design, or ML hardware acceleration.
NetflixNASDAQ: NFLX: Global video streaming and media production service.
7+ YOE7+ years software engineering; 3+ years ML infrastructure, model serving, or ML platform experience in ads/real-time decisioning; real-time model serving with sub-20ms latency; proficiency in Java, Python, or Scala; experience with ML serving frameworks and real-time feature pipelines; strong model monitoring and production readiness.
Java, Python, Scala, ML serving frameworks, feature stores, model registries
San Jose or Durham or Mexico City or Vancouver or Bengaluru or Pune or Hoofddorp or Belgrade or Barcelona or Singapore or Sydney or Tokyo
$171k-$257k/yrHybridFull Time
NutanixNASDAQ: NTNX: Sells cloud software and hyperconverged infrastructure for enterprises.
8+ YOE8+ years building distributed, high-performance systems; strong Go/Python, Docker, Kubernetes, CI/CD; knowledge of datacenter, OS internals, virtualization, and ML frameworks.
Sr. Staff AI Engineer - On-Prem AI Infrastructure & Agentic Systems
San Jose, California, United States
$140k-$165k/yrOnsiteFull Time
SK hynix Memory Solutions AmericaKorea Exchange: 000660: Develops semiconductor controllers and firmware for enterprise data storage.
2+ YOE2+ years in AI/ML engineering with on-prem/private-cloud deployment, experience building agentic AI and RAG pipelines, model fine-tuning (LoRA/QLoRA), Python/Linux/Docker/Kubernetes proficiency, and familiarity with vector DBs and AI serving frameworks.
Ensemble Health Partners: Manages financial operations and technology for healthcare organizations.
3+ YOEBachelor's in CS/Engineering/Data Science/Mathematics required; Python and ML/AI production experience; 3+ years relevant experience typical; experience with deep learning frameworks and LLM techniques preferred.