317 ai framework engineer jobs at 177 companies in Pleasanton, CA

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
1mo
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
3mo
Save
Mark Applied
Hide
Staff AI Engineer
San Francisco, California, United States
$182k-$310k/yr HybridFull Time
Airwallex
Airwallex: Global financial platform for business payments and money management.
5+ YOE5+ years in applied AI/ML engineering with production deployments; strong Python; experience with ML frameworks (PyTorch, JAX, TensorFlow); track record deploying AI agents or orchestration systems; knowledge of monitoring and iteration in production AI systems.
Python, PyTorch, JAX, TensorFlow, LangChain, LlamaIndex, AutoGen, CrewAI, Semantic Kernel
2mo
Save
Mark Applied
Hide
AI Engineer
San Francisco, California, United States
$150k-$210k/yr OnsiteFull Time
Cargomatic
Cargomatic: Digital marketplace connecting shippers with local trucking capacity.
3+ YOE3+ years software engineering with focus on AI/ML or automation; hands-on AI-powered applications; experience with agentic AI concepts, LLMs, and modern AI frameworks; cloud infrastructure (AWS); backend (Node.js); frontend (React).
Claude, OpenAI, LangChain, n8n, AWS, Node.js, React, MongoDB, PostgreSQL, MySQL, S3, ECS, Lambda, REST APIs, GitHub
2mo
Save
Mark Applied
Hide
Senior ML/AI Engineer
San Francisco or New York
OnsiteFull Time
Beast Industries
Beast Industries: Produces digital media and consumer goods for MrBeast brands.
5+ YOESenior ML/AI engineer with production experience, AI-native mindset, strong eval framework, and ability to ship features at consumer scale.
Python, Machine Learning, Model Serving, Inference Infrastructure, Retrieval-Augmented-Generation, Agent Frameworks, Evaluation Systems, Prompt Engineering, Coding Agents
1mo
Save
Mark Applied
Hide
Applied AI Engineer
San Mateo, California, United States
OnsiteFull Time
Sol
Sol: AI-native HRIS platform for modern workforce management.
Experience building production AI systems and infrastructure, designing evaluation frameworks, prioritizing explainability, privacy and compliance, and making model/tooling tradeoff decisions.
3mo
Save
Mark Applied
Hide
AI Systems Performance Engineer
San Jose, California, United States
$141k-$226k/yr OnsiteFull Time
Broadcom
BroadcomNASDAQ: AVGO: Designs and sells semiconductors and infrastructure software to enterprises.
12+ YOESenior AI fabric performance engineer with Linux, Python/C++, ML frameworks, and Ethernet fabric optimization; 12+ years experience.
Linux, Python, C++, PyTorch, NCCL, MLPerf, RDMA, RoCEv2, Docker, Kubernetes
2d
Save
Mark Applied
Hide
AI/ML Enablement Lead Engineer
New York or Atlanta or Hopkins or Minneapolis or San Francisco or Cupertino or Charlotte or Chicago or Irving
$148k-$175k/yr HybridFull Time
U.S. Bank
U.S. BankNYSE: USB: Provider of personal, business, and institutional financial services.
Strong software engineering background and experience designing, building, or delivering AI/ML solutions (LLM, RAG, Agentic AI, GenAI). Knowledge of orchestration frameworks, enterprise AI platforms, solution architecture, prompt engineering, and stakeholder leadership.
LangChain, LangGraph, Semantic Kernel, MCP, Azure OpenAI, OpenAI, Anthropic, Gemini
1mo
Save
Mark Applied
Hide
Senior Deep Learning Framework Communications Engineer
Santa Clara or Austin or Westford or Durham or United States
$152k-$288k/yr HybridFull Time
NVIDIA
NVIDIANASDAQ: NVDA: Designs graphics processing units and artificial intelligence hardware.
5+ YOE5+ years software engineering experience in HPC/AI, experience with PyTorch/JAX and inference engines, Python/C++/CUDA development, performance benchmarking and profilers, understanding of multi-GPU communication and compilers.
PyTorch, TRT-LLM, vLLM, SGLang, JAX, NCCL, NVSHMEM, GPUDirect, torch.compile, PyTorch profiler, NVIDIA Nsight Systems, Python, C++, CUDA, Triton, cuTe, MPI
3d
Save
Mark Applied
Hide
Senior AI Engineer
San Francisco, California, United States
HybridFull Time
H2O.ai
H2O.ai: Provides open-source machine learning and predictive analytics for enterprises
8+ YOE8+ years AI/ML engineering experience, enterprise delivery leadership, LLM and agentic AI experience, strong Python and ML framework skills, cloud and Kubernetes deployment experience.
Python, PyTorch, TensorFlow, scikit-learn, LangChain, LlamaIndex, AWS, Azure, GCP, Kubernetes, Docker, Wave, H2O Document AI
3mo
Save
Mark Applied
Hide
Senior Embodied AI Engineer
Milpitas, California, United States
OnsiteFull Time
UnitX
UnitX: Builds AI-powered robotics for automated visual inspection in manufacturing.
MS or PhD in CS, Robotics, ML, AI or related field; strong Python and C++; experience with ML frameworks and robotics simulation.
Python, C++, PyTorch, JAX, TensorFlow, NVIDIA Isaac Sim, MuJoCo, Drake, CUDA, Jetson, Vision-Language-Action, USD, URDF, MJCF
1mo
Save
Mark Applied
Hide
Senior Principal Engineer- End-to-End AI Training Framework
Sunnyvale, California, United States
$240k-$320k/yr HybridFull Time
Bosch
Bosch: Global manufacturer of automotive and industrial engineering technology.
10+ YOE10+ years in software and system engineering for autonomous driving; advanced degree in CS/Robotics/EE/AI; strong AI stack, cloud training, MLOps, and production deployment experience.
TensorFlow, PyTorch, Python, C++, MLOps, CI/CD, SIL, HIL, Embedded tools
1w
Save
Mark Applied
Hide
AI Engineer
San Francisco or Seattle or Dallas or Austin or Chicago or Atlanta or New York or San Diego
$100k-$120k/yr HybridFull Time
Jacobs
JacobsNYSE: J: Global provider of professional engineering and technical services.
1+ YOE1-3 years software development experience, bachelor's in CS/AI/data science, proficiency with AI/LLM APIs and frameworks, Python and JavaScript/TypeScript skills, Git and CI/CD experience, Docker/Podman familiarity.
Python, TypeScript, JavaScript, LangChain, LangGraph, CrewAI, PydanticAI, Model Context Protocol (MCP), Anthropic Claude API, OpenAI API, OpenAI Codex, Claude, Cursor, GitHub, GitHub Copilot, SQL, Docker, Podman, React, HTML, CSS, FastAPI, Flask, Express, Pinecone, Weaviate, pgvector, Microsoft Azure, Microsoft Fabric, Google Cloud, Amazon Web Services, Power BI
1mo
Save
Mark Applied
Hide
AI Infrastructure Engineer
San Jose, California, United States
$179k-$306k/yr HybridFull Time
AMD
AMDNASDAQ: AMD: Designs and manufactures computer processors and graphics technology.
5+ YOE5+ years in DevOps/platform/infrastructure engineering; deep Kubernetes experience; experience building developer-facing platforms, Helm and GitOps workflows, storage/networking for GPU workloads; Terraform, monitoring, and ML framework exposure preferred.
Kubernetes, Helm, ArgoCD, Flux, GitOps, Terraform, CSI drivers, CNI, Prometheus, Grafana, Loki, Slurm, PyTorch, vLLM, SGLang
1d
Save
Mark Applied
Hide
AI Engineer (US)
New York or Philadelphia or San Francisco
OnsiteFull Time
Lynx Analytics
Lynx Analytics: Provides graph-based AI and big data analytics solutions for enterprises.
5+ YOE5–8 years software/ML engineering experience with 2–3 years building LLM/agentic AI systems; hands-on with agent frameworks, vector DBs, knowledge-graph retrieval, Python, and production engineering practices.
LangChain, LlamaIndex, AutoGen, CrewAI, OpenAI, Anthropic, Neo4j, Microsoft GraphRAG, Pinecone, Weaviate, Qdrant, Python, Docker, Kubernetes, CI/CD
3w
Save
Mark Applied
Hide
AI Engineer
San Mateo, California, United States
$200k-$290k/yr OnsiteFull Time
P-1 AI
P-1 AI: Developing AI agents for industrial engineering and physical design.
Experience shipping AI research to production, deep learning and LLM expertise, fluent Python, familiarity with agentic frameworks (e.g., LangChain), and experience with physical systems (mechanical/robotics).
Python, LangChain
2mo
Save
Mark Applied
Hide
AI Engineer - Model Performance
San Francisco, California, United States
HybridFull Time
Fathom
Fathom: AI assistant for recording, transcribing, and summarizing virtual meetings.
Deep experience with LLM serving frameworks, quantization, and production fine-tuning; strong Python and GPU performance analysis.
Python, PyTorch, vLLM, SGLang, TensorRT-LLM, LoRA, QLoRA, torchtune, Axolotl
2mo
Save
Mark Applied
Hide
Sr AI Automation Engineer
Austin or New York City or San Francisco or Arlington
HybridFull Time
BetterUp
BetterUp: Digital platform for professional coaching and employee mental fitness.
7+ YOE7+ years of engineering experience shipping real systems at scale; experience with agent frameworks, Claude SDK, MCP; cross-functional collaboration; ownership mindset.
Claude, Anthropic SDK, MCP, Slack, Jira, Salesforce, Okta, Workday, Google Workspace, AWS
3d
Save
Mark Applied
Hide
Principal Engineer, AI Compiler
San Jose, California, United States
$150k-$230k/yr OnsiteFull Time
Renesas Electronics
Renesas ElectronicsTokyo Stock Exchange: 6723: Designs and manufactures semiconductors for automotive and industrial systems.
MS/PhD or equivalent experience; deep experience building AI compilers, accelerator backends or graph optimization frameworks; expertise with MLIR, LLVM, TVM-style systems; strong C/C++ and Python skills; NPU/performance optimization experience.
MLIR, LLVM, TVM, C, C++, Python
1mo
Save
Mark Applied
Hide
Senior Staff AI Software Engineer
San Jose, California, United States
$189k-$301k/yr OnsiteFull Time
Samsung Semiconductor
Samsung SemiconductorKorea Exchange: 005930: Designs and manufactures memory chips, processors, and sensors.
10+ YOEExtensive industry experience (10+ years), high-performance AI framework development for GPUs/accelerators, PyTorch and vLLM expertise, deep knowledge of LLMs, memory architecture (HBM), Linux, and tooling (GitHub, Jira).
PyTorch, vLLM, Linux, GitHub, Jira, GPUs, High Bandwidth Memory (HBM)
2mo
Save
Mark Applied
Hide
AI engineer
San Francisco or New York City
$152k-$316k/yr HybridFull Time
Writer
Writer: Platform for building and deploying enterprise generative AI agents.
5+ YOE5+ years of experience building and deploying ML/AI systems; proficient in Python and frameworks (PyTorch, TensorFlow, or JAX); strong ML/natural language processing background; experience with cloud (AWS, GCP, Azure) and MLOps; able to deliver scalable software; strong collaboration and problem-solving.
Python, PyTorch, TensorFlow, JAX, AWS, GCP, Azure, MLOps tools