HiringCafe
Join our community

308 ai framework engineer jobs at 173 companies in Berkeley, 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
Help us #BeatIndeed & #BeatLinkedIn
1w
Save
Mark Applied
Hide
AI Engineer
Cupertino, California, United States
$70k-$130k/yr OnsiteFull Time
Tata Consultancy Services
Tata Consultancy ServicesNational Stock Exchange of India: TCS: Global provider of IT services, consulting, and business solutions.
3+ YOE3–6 years ML/AI experience, strong Python, ML frameworks (PyTorch, TensorFlow, scikit-learn), data pipelines, LLM experience (LangChain/LangGraph), experiment tracking (MLflow, Weights & Biases), and software engineering fundamentals.
Python, PyTorch, TensorFlow, scikit-learn, LangChain, LangGraph, MLflow, Weights & Biases, Git, DVC, Docker, API
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
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
Job PostingView all
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
Job PostingView all
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
Job PostingView all
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.
Job PostingView all
2mo
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
Job PostingView all
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
Job PostingView all
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
Job PostingView all
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
Job PostingView all
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
Job PostingView all
2w
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
Job PostingView all
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
Job PostingView all
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
Job PostingView all
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)
Job PostingView all
1mo
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
Job PostingView all
2mo
Save
Mark Applied
Hide
Senior AI Frameworks Engineer
Santa Clara, California, United States
$152k-$288k/yr OnsiteFull Time
NVIDIA
NVIDIANASDAQ: NVDA: Designs GPU-accelerated computing and artificial intelligence hardware.
3+ YOEMS or PhD in CS/EE; 3+ years; strong Python and C++, API design, library/framework development; experience delivering high-performance extensions.
Python, C++, LLVM/MLIR, FFI
Job PostingView all
11h
Save
Mark Applied
Hide
AI Engineer
San Francisco, California, United States
OnsiteFull Time
Corridor
Corridor: Secures software codebases using AI-powered autonomous vulnerability detection.
2+ YOE2+ years industry or research experience in software engineering/ML, production-quality coding in Python, familiarity with ML frameworks, vector stores, hybrid search, and databases; experience with model fine-tuning and evaluation pipelines preferred.
Python, ML frameworks, vector stores, hybrid search, databases
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
1mo
Save
Mark Applied
Hide
Principal AI Engineer
Santa Clara or Milpitas or Mountain View or East Foothills or Los Altos or Stanford
$176k-$293k/yr HybridFull Time
BMC Software
BMC Software: Provides enterprise software for IT management and automation.
8+ YOE8+ years building and operating production-grade software with recent depth in AI/ML; proven experience shipping agentic AI systems, evaluation/experimentation frameworks, RAG/retrieval, and cloud-native tooling (Git, Docker, Kubernetes, CI/CD, AWS/Azure/GCP).
LangGraph, Google ADK, Git, Docker, Kubernetes, CI/CD, AWS, Azure, GCP
Job PostingView all

Explore Jobs

Expand Your Job Search