HiringCafe
Join our community

758 ai platform engineer jobs at 322 companies in Berkeley, CA

PromotedHiringCafe
ML Engineer - Inference & Model Deployment
Cupertino, CA, US
$250k-$310k/yr On-SiteFull Time
HiringCafe
HiringCafe: Building a 100× better job search engine to take on Indeed and LinkedIn.
Turn powerful AI and ML models into fast, reliable production systems. Own inference latency, throughput, model-serving architecture, multi-GPU systems, and production deployment for millions of users.
Python, PyTorch, vLLM, SGLang, TensorRT, LLMs
Help us #BeatIndeed & #BeatLinkedIn
PromotedHiringCafe
Founding Machine Learning / AI Search Engineer
Cupertino, CA, US
$160k-$310k/yr On-SiteFull Time
HiringCafe
HiringCafe: Building a 100× better job search engine to take on Indeed and LinkedIn.
Build the ML and AI search behind HiringCafe — ranking, recommenders, retrieval, and LLM agents that surface jobs people would never find on their own.
Python, PyTorch, Elasticsearch, LLMs
Help us #BeatIndeed & #BeatLinkedIn
PromotedHiringCafe
Founding Backend / Infra Engineer
Cupertino, CA, US
$160k-$300k/yr On-SiteFull Time
HiringCafe
HiringCafe: Building a 100× better job search engine to take on Indeed and LinkedIn.
Own the crawlers, pipelines, and infrastructure powering a real-time job search engine. Strong Node.js and Python fundamentals; bonus points for security and reverse-engineering chops.
Node.js, Python, Elasticsearch, Redis
Help us #BeatIndeed & #BeatLinkedIn
2mo
Save
Mark Applied
Hide
Principal AI Platform Engineer
San Francisco or Toronto or Montreal or United States or Canada
RemoteFull Time
MaintainX
MaintainX: Provides mobile-first software for maintenance and asset management.
10+ YOE10+ years software engineering experience, 3+ years building ML/LLM systems, strong Python backend skills, platform engineering and developer-experience experience, technical leadership at Principal level.
Python, MCP, LangGraph, LlamaIndex, Langfuse, LangSmith, Braintrust
Job PostingView all
1mo
Save
Mark Applied
Hide
Senior AI Platform Engineer - Frisco
Frisco or San Jose
$107k-$176k/yr HybridFull Time
McAfee
McAfee: Provides cybersecurity and privacy protection software for consumers.
10+ YOE10+ years platform engineering experience with GenAI/LLM ecosystems, Kubernetes (EKS/GKE), cloud (AWS/GCP), Python and Go, Terraform, Backstage, CI/CD, vector stores/OpenSearch, Kafka, and enterprise AI security and governance.
AWS Bedrock, OpenAI, Anthropic, Kubernetes, GKE, EKS, Python, Go, FastAPI, gRPC, AWS, GCP, Terraform, Backstage, OpenSearch, Aurora pgvector, CI/CD, Apache Kafka, LangChain, LlamaIndex, Pinecone, Weaviate, IAM, VPC, KMS, MLOps, LLMOps
Job PostingView all
4d
Save
Mark Applied
Hide
AI Platform Engineer
Milpitas, California, United States
$136k-$232k/yr OnsiteFull Time
KLA
KLANASDAQ: KLAC: Provides process control and yield management for semiconductor manufacturing.
5+ YOEDegree in CS/CE or related, 5+ years systems/DevOps/ML infrastructure experience, hands-on AI/GPU cluster and Linux/Kubernetes expertise, Python/Bash scripting, and knowledge of storage, networking, and security.
Linux, Kubernetes, Docker, Slurm, Ray, TensorFlow, PyTorch, GPU, TPU, Bash, Python, CI/CD, Infrastructure as Code (IaC), TPM
Job PostingView all
1w
Save
Mark Applied
Hide
Lead AI Platform Engineer
Madison or San Diego or Redwood City
$151k-$256k/yr OnsiteFull Time
Exact Sciences
Exact SciencesNASDAQ: EXAS: Provides molecular diagnostic tests for early cancer detection.
3+ YOE3+ MgmtDesign and deploy AI/ML and NLP solutions (including LLM/Gen AI) with leadership experience, strong Python skills, AI/ML knowledge, ethical AI awareness, and 3+ years project leadership experience.
Python, Scaled Agile Frameworks (SAFE)
Job PostingView all
1mo
Save
Mark Applied
Hide
Principal AI Platform Engineer
San Francisco, California, United States
RemoteFull Time
Nextdata
Nextdata: Provides an operating system for building autonomous data products.
Extensive experience in data engineering, AI-enabled data systems, distributed systems, and governance; strong Python, SQL, API, and systems skills; experience with agentic interfaces, retrieval/semantic systems, and building production-safe data products.
Python, SQL, MCP, Databricks, Snowflake, BigQuery, Spark, DuckDB, Postgres, vector databases, graph databases, SSO, OAuth, OIDC, SAML, SCIM, RBAC, ABAC
Job PostingView all
4d
Save
Mark Applied
Hide
Senior Platform AI Engineer
Santa Clara or California
$184k-$357k/yr HybridFull Time
NVIDIA
NVIDIANASDAQ: NVDA: Designs graphics processing units and artificial intelligence hardware.
8+ YOE8+ years building and operating production platform/backend infrastructure; 5+ years ML infrastructure; strong Python and a compiled language; experience with job queues, sandboxed execution, and production reliability.
Python, C, C++, Go, Java, Rust, Kubernetes Jobs, Celery, Sidekiq, Temporal, container runtimes
Job PostingView all
1w
Save
Mark Applied
Hide
Staff AI Platform Engineer
San Francisco or Alabama or Arkansas or California or Colorado or Florida or Georgia or Illinois or Kentucky or Massachusetts or Michigan or Montana or Missouri or Nevada or New Jersey or New York or North Carolina or Oregon or Pennsylvania or Texas or Washington or Utah or Virginia or Wisconsin or United States
$169k-$273k/yr HybridFull Time
Afresh
Afresh: AI-powered inventory management for grocery store fresh food departments.
5+ YOE5+ years building production software/data/ML systems; strong Python, systems and API design; hands-on LLM systems (retrieval/RAG, agents, evaluation); Databricks/MLflow and modern data-platform experience.
Python, PySpark, dbt, Databricks, Delta Lake, Unity Catalog, MLflow, Astronomer, Airflow, Claude, pgvector, Pinecone, Weaviate, MCP, LangGraph, GitHub, Shortcut, Notion
Job PostingView all
4d
Save
Mark Applied
Hide
Senior Platform AI Engineer
Santa Clara or United States
$184k-$357k/yr HybridFull Time
NVIDIA
NVIDIANASDAQ: NVDA: Designs GPU-accelerated computing and artificial intelligence hardware.
8+ YOE8+ years designing and operating production platform/backend infrastructure,5+ years ML infrastructure,BS/MS/PhD or equivalent in CS/EE/CE,strong Python and compiled-language skills,experience with Kubernetes Jobs and job queues.
Python, C, C++, Go, Java, Rust, Kubernetes Jobs, Celery, Sidekiq, Temporal, container runtimes, EDA
Job PostingView all
2mo
Save
Mark Applied
Hide
AI Platform Engineer, Training and Inference
San Francisco, California, United States
HybridFull Time
Saviynt
Saviynt: Provides AI-powered identity governance and cloud security platforms.
ML platform or MLOps engineer with production Ray experience; LLM serving, distributed training, Python and PyTorch; MLflow/Flyte; Bachelor's degree in CS/Engineering.
Ray Train, Ray Serve, Ray Core, Ray Data, vLLM, SGLang, NVIDIA Triton, TorchTrainer, DDP, NCCL, PPO, RLlib, Flyte, MLflow, Qdrant, Pgvector, PyTorch, Python
Job PostingView all
1mo
Save
Mark Applied
Hide
AI Platform Engineer
San Mateo, California, United States
$170k-$205k/yr HybridFull Time
Notable
Notable: Automates administrative and clinical workflows for healthcare providers using AI.
Production Python proficiency, experience building and operating LLM-powered features, familiarity with Jupyter environments, designing for reliability/scalability/observability, and ability to decompose ambiguous problems and own delivery end-to-end.
Python, Jupyter Notebook, JupyterLab, Databricks, Colab, LLM, Terraform, Helm Charts, Google Cloud Platform, GKE, Cloud Run, Cloud Storage, TypeScript, Postgres
Job PostingView all
3w
Save
Mark Applied
Hide
Distinguished AI Engineer (Agentic AI Platform)
San Francisco or McLean or Cambridge or San Jose or New York City
$294k-$335k/yr OnsiteFull Time
Capital One
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.
Python, Go, Scala, Java, C++, C#, Golang, LangGraph, AutoGen, Semantic Kernal, Semantic Kernel (Microsoft), CrewAI, LlamaIndex, LangChain, AWS, Google Cloud, Azure, Google Cloud Vertex AI, Amazon SageMaker, Azure Machine Learning, K8s, Helm charts, CRDs, SDK, CLI, VectorDBs, RAG, LLM Inference, Similarity Search
Job PostingView all
3w
Save
Mark Applied
Hide
Senior AI Platform Engineer
San Francisco or Boston or Jacksonville or New York or Windsor
$140k-$180k/yr HybridFull Time
SS&C Technologies
SS&C TechnologiesNasdaq: SSNC: Provides software and technology for financial and healthcare sectors.
5+ YOE5+ years Python experience; data engineering/distributed systems (Spark, Hive, Trino, StarRocks, DBT, Airflow, Kubernetes); LLM and agent frameworks (LangChain, LlamaIndex, DSPy, OpenAI MCP, A2A); API design (REST, GraphQL, gRPC); cloud/containerization and enterprise security.
Python, Spark, Hive, Trino, StarRocks, DBT, Airflow, Kubernetes, Docker, LangChain, LlamaIndex, DSPy, OpenAI MCP, A2A, REST, GraphQL, gRPC, pgvector, Pinecone, Weaviate, Milvus, Vespa, Kafka, Pulsar, OpenTelemetry, DataHub, MLflow, Prometheus, Hugging Face
Job PostingView all
1w
Save
Mark Applied
Hide
AI/ML Platform Engineer
Santa Clara, California, United States
$179k-$306k/yr OnsiteFull Time
AMD
AMDNASDAQ: AMD: Designs and manufactures computer processors and graphics technology.
Experienced systems engineer to build and operate distributed ML platform infrastructure, GPU cluster scheduling, experiment tracking, and production developer tooling. Bachelor's required or equivalent; Master's/PhD preferred.
Python, C++, Go, Rust, ROCm/HIP, CUDA, vLLM, SGLang, Triton, PyTorch, JAX, Ray, Kubernetes, Slurm, MLflow, Weights & Biases, CI/CD
Job PostingView all
2mo
Save
Mark Applied
Hide
Staff Engineer, AI Platform
Santa Monica or New York or San Francisco or Los Angeles
$200k-$220k/yr HybridFull Time
Pivotal Health
Pivotal Health: AI platform automating healthcare insurance claim disputes for providers.
Hands-on software engineering leader building platform infrastructure for AI systems; experience with agent runtimes, observability, experimentation, and internal tooling; strong API/SDK design; comfortable in greenfield work; healthcare/payments domain a plus.
Python, FastAPI, GCP, LLM platforms, Observability tooling, A/B testing, SDKs and APIs
Job PostingView all
1mo
Save
Mark Applied
Hide
AI Platform Architect
Austin or Milpitas
OnsiteFull Time
Graphcore
Graphcore: Produces specialized processors for artificial intelligence workloads.
4+ YOELead/Principal architect for large-scale AI platforms; 4+ years in distributed AI; strong knowledge of workloads, interconnects, and orchestration.
Kubernetes, Slurm, PyTorch, DeepSpeed, RDMA, PCIe, NVMe, Numa, Docker, Infrastructure as Code
Job PostingView all
1mo
Save
Mark Applied
Hide
Platform Engineer (SRE) - AI Control Plane
San Francisco, California, United States
OnsiteFull Time
Speakeasy
Speakeasy: Automates API SDK and documentation generation for developers.
Platform Engineer (SRE) to own reliability, design deployments, and participate in on-call; strong systems and software engineering.
Job PostingView all
1mo
Save
Mark Applied
Hide
Platform Engineer, AI Tooling
San Francisco, California, United States
$132k-$179k/yr HybridFull Time
Drata
Drata: Automated security and compliance platform for businesses.
3+ YOE3+ years software engineering; experience with AI tooling and developer platforms; MCP and CI/CD; AWS; TypeScript or Python; CI/CD tooling; knowledge of Anthropic API or equivalents; strong prompt engineering; agile experience.
MCP, TypeScript, Python, REST, AWS, GitHub Actions, Temporal, MySQL, Anthropic API, CI/CD, Node.js
Job PostingView all
1mo
Save
Mark Applied
Hide
Lead AI Platform Engineer
Chicago or Hopkins or Minneapolis or Cincinnati or Cupertino or Charlotte
$139k-$164k/yr HybridFull Time
U.S. Bank
U.S. BankNYSE: USB: Provider of personal, business, and institutional financial services.
6+ YOESenior technical role requiring 6+ years relevant experience designing, building, and operationalizing GenAI solutions using LLMs, RAG, cloud platforms, containerization, IaC, and Python; strong leadership and architecture skills.
LangChain, LangGraph, AWS Bedrock, Microsoft Foundry Agent Service, Microsoft Azure, AWS, Docker, Kubernetes, Terraform, Microsoft ARM, Microsoft Bicep, Python
Job PostingView all
2mo
Save
Mark Applied
Hide
AI Platform Security Engineer
San Jose, California, United States
OnsiteFull Time
Kai Cyber
Kai Cyber: Agentic AI platform for autonomous enterprise cybersecurity operations.
6+ YOE6+ years in cloud/security engineering; expert Azure security knowledge; Kubernetes security; IaC security; scripting; experience with AI/ML data security.
Azure, Azure Entra ID, Azure Key Vault, Azure Policy, Azure Networking, AKS, Terraform, Bicep, Notary, Cosign, PIM, CI/CD tooling, Python, Bash, PowerShell
Job PostingView all
1mo
Save
Mark Applied
Hide
AI Platform Engineer
Lincoln or San Francisco
$125k-$165k/yr RemoteFull Time
TELCOR
TELCOR: Provides healthcare software for laboratory and point-of-care operations.
2+ YOE2+ years experience with TypeScript and ReactJS; experience with LLMs; Redis and Postgres; personal projects; open source contributions preferred.
TypeScript, ReactJS, LLMs, Redis, Postgres
Job PostingView all

Explore Jobs

Expand Your Job Search