Skip to content
2,634 jobs

Job results

Together AI

Together AI

San Francisco, CA

Machine Learning Engineer - Inference
$160k+/yrOn-site3+ YOEML Engineering

Optimizes and builds production inference systems for large language models at scale using PyTorch and high-performance tooling. Requires 3+ years experience in production code, OS concepts, and AI inference systems.

Together AI

Together AI

San Francisco, CA

Systems Research Engineer, GPU Programming
$160k+/yrRemoteAI Research

Develops and optimizes GPU-accelerated kernels and algorithms for ML/AI applications, co-designing with modeling, hardware, and software teams. Requires strong GPU programming expertise in CUDA/Triton and knowledge of ML models.

Databricks

Databricks

San Francisco, CA
Staff Machine Learning Engineer
$190k+/yrHybridML Engineering

Develops and deploys state-of-the-art GenAI models and systems for Databricks products like Assistant and Genie. Requires 2-8 years ML engineering experience, proficiency in Python/PyTorch/TensorFlow, and expertise in LLMs.

Maven Robotics

Maven Robotics

San Francisco, CA

Machine Learning Engineer - Robot Perception
No salary listedOn-siteML Engineering

Develops and deploys ML-based perception algorithms for robot object detection, pose estimation, tracking, and scene understanding. Integrates sensor fusion from cameras, LiDAR, IMUs; requires MS/PhD in ML/CS/robotics, expertise in computer vision, PyTorch/TensorFlow, and real-world robotics hardware.

Maven Robotics

Maven Robotics

San Francisco, CA

Machine Learning Engineer - Robot Manipulation
No salary listedOn-siteML Engineering

Designs and deploys reinforcement and imitation learning algorithms for robotic manipulation tasks in dynamic environments. Requires MS/PhD, deep RL/IL expertise, PyTorch proficiency, and real-world ML deployment experience in a fast-paced startup.

Sema4.ai

Sema4.ai

Atlanta, GA
Staff Engineer, AI
No salary listedOn-site7+ YOEML Engineering

Designs, builds, and owns core cognitive architectures for enterprise AI agent platform, blending AI research with production systems. Requires 7+ years engineering with 2+ years AI/ML production experience and deep Python backend expertise.

Databricks

Databricks

Mountain View, CA
Senior Manager, Infrastructure Data Science
$229k+/yrOn-site10+ YOEData Science

Leads a team of data scientists to deliver data-driven solutions for infrastructure capacity planning, performance optimization, reliability, and efficiency at Databricks. Requires 10+ years in infrastructure data science/ML and 5+ years management experience.

Harvey

Harvey

San Francisco, CA
Mid/Senior/Staff Software Engineer, Agents
$165k+/yrHybridML Engineering

Build AI agent systems for legal workflows, optimizing performance via prompt engineering, model selection, tools, and evals. Requires 3+ years experience, Python proficiency, and LLM/agent framework expertise for mid/senior/staff levels.

Databricks

Databricks

San Francisco, CA

Staff Software Engineer - GenAI Performance and Kernel
$191k+/yrOn-siteML Engineering

Designs, implements, and optimizes high-performance GPU kernels for GenAI inference stack. Leads performance improvements, mentors engineers, and collaborates with ML and systems teams. Requires deep kernel programming and GPU architecture expertise.

Databricks

Databricks

San Francisco, CA

Staff Software Engineer - GenAI inference
$191k+/yrOn-site6+ YOEML Engineering

Leads architecture, development, and optimization of GenAI inference engine for high-throughput, low-latency LLM serving. Requires 6+ years in performance-critical systems, deep ML inference expertise, CUDA/GPU programming, and distributed systems.

Databricks

Databricks

San Francisco, CA

Staff Software Engineer, Model Serving
$192k+/yrOn-site10+ YOEML Engineering

Designs and builds scalable, low-latency model serving infrastructure for AI/ML models across CPU/GPU workloads. Requires 10+ years in large-scale distributed systems and deep expertise in inference systems, architecture, and cross-team collaboration.

Databricks

Databricks

San Francisco, CA

Staff Software Engineer, Foundational Model Serving
$192k+/yrOn-site10+ YOEML Engineering

Designs and builds scalable, low-latency systems for serving frontier AI models on GPUs. Requires 10+ years in large-scale distributed systems, strong system design skills, and leadership in operational excellence; no prior AI experience needed.

Databricks

Databricks

San Francisco, CA

Senior Software Engineer, Model Serving
$166k+/yrOn-site5+ YOEML Engineering

Designs and builds scalable infrastructure for high-throughput, low-latency AI/ML model serving on CPU/GPU. Requires 5+ years in distributed systems, inference expertise, and strong system design skills.

Databricks

Databricks

San Francisco, CA

PhD GenAI Research Scientist Intern
$54+/hrOn-siteAI Research

PhD intern researches and develops techniques to adapt LLMs and AI systems for enterprise domains, including method design, evaluation, and efficient post-training. Requires deep learning proficiency, PyTorch skills, and ongoing PhD studies.

Databricks

Databricks

Bellevue, WA

Sr. Staff Software Engineer - Unity Catalog Data Governance
$217k+/yrOn-site15+ YOEData Engineering

Leads technical direction and development of Unity Catalog's governance features for secure data and AI asset management at scale. Requires 15+ years in large-scale distributed systems, deep CS expertise, and strong leadership.

Databricks

Databricks

Bellevue, WA
Staff Software Engineer - Distributed Data Systems
$192k+/yrOn-site8+ YOEData Engineering

Develops distributed data systems like Apache Spark and Delta Lake at massive scale, ensuring high performance and reliability for exabyte-scale workloads. Requires 8+ years in Java/Scala/C++ and deep distributed systems expertise.

Databricks

Databricks

Mountain View, CA

Staff Data Scientist - Infrastructure
$192k+/yrOn-site7+ YOEData Science

Leads data science initiatives to inform business decisions, generate strategic insights for engineering priorities, and build production ML tooling. Requires 7+ years experience, strong Python/SQL/Spark skills, and MS/PhD in quantitative field.

Databricks

Databricks

San Francisco, CA

Staff Data Scientist - Trust and Safety
$192k+/yrOn-site7+ YOEData Science

Develops ML models for fraud/abuse detection and anomalous activity on Databricks platform. Analyzes security features, collaborates cross-functionally, and deploys production solutions. Requires 7+ years experience, MS in quantitative field, Python/SQL/Spark expertise.

Databricks

Databricks

Mountain View, CA

Senior Software Engineer - Distributed Data Systems
$166k+/yrOn-site5+ YOEData Engineering

Develop distributed data systems including Apache Spark and Delta Lake to handle big data workloads efficiently. Requires 5+ years in Java/Scala/C++ and expertise in distributed systems.

Databricks

Databricks

Bellevue, WA
Senior Software Engineer - Distributed Data Systems
$166k+/yrOn-site5+ YOEData Engineering

Senior engineer building distributed data systems like Apache Spark and Delta Lake to handle big data processing, ETL, and data science workloads. Requires 5+ years in Java/Scala/C++ and expertise in distributed systems.

Databricks

Databricks

San Francisco, CA

Staff Data Scientist
$192k+/yrOn-site7+ YOEData Science

Staff Data Scientist drives data-driven decisions through segmentation, recommendations, forecasting, and product analytics. Collaborates cross-functionally, mentors juniors, and requires 7+ years experience with Python/Scala, Spark, SQL, and MS/PhD.

Chess.com

Chess.com

United States

Data Analyst, Growth & Monetization - ChessKid
No salary listedRemote3+ YOEData Analytics

Analyze player behavior data for ChessKid to drive growth and monetization through insights on retention, engagement, and experiments. Collaborate with product and business teams using Amplitude, BigQuery, SQL, and Python.

Astera

Astera

Senior Software Engineer
$100k+/yrHybridML Engineering

Builds high-performance infrastructure for brain reverse-engineering via large-scale AI models and neuroscience-informed simulations. Architects systems, optimizes RL environments, and scales research prototypes to production.

Airbnb

Airbnb

United States

Staff GenAI System Engineer, Automation Foundation
$204k+/yrRemote9+ YOEML Engineering

Leads development of Airbnb's conversational AI Automation Platform and agent provisioning systems. Drives backend optimization, prompt engineering, and LLM integrations with 9+ years experience in scalable architectures.

Clarium

Clarium

United States

Forward Deployed Engineer (Data)
No salary listedRemote4+ YOEData Engineering

Builds and owns customer data integrations for hospital systems, designing production ETL pipelines and reusable connectors. Requires 4+ years experience with Python, SQL, AWS, and direct customer collaboration in ambiguous environments.

Sentry

Sentry

San Francisco, CA

Senior Software Engineer, AI Evals
$240k+/yrHybrid5+ YOEML Engineering

Build evaluation infrastructure for AI systems at Sentry, designing datasets, benchmarks, and test harnesses to measure accuracy and reliability of debugging agents. Requires 5+ years experience, Python/TypeScript proficiency, and AI/ML background.

Agentio

Agentio

New York, NY

Marketing Science Lead
$180k+/yrOn-site7+ YOEData Analytics

Leads establishment of measurement practice for creator marketing, defining ROI frameworks, advising top brands on MMM/incrementality/attribution, and driving product roadmap for scalable analytics.

Cursor

Cursor

San Francisco, CA
Software Engineer, ML Infrastructure
No salary listedOn-siteML Engineering

Builds large-scale ML infrastructure including GPU clusters, training frameworks, and workload schedulers. Requires strong systems engineering in Python, Rust, Golang, with Kubernetes and distributed systems experience.

Headway

Headway

New York, NY
Staff Data Scientist - Product Analytics
$212k+/yrRemote10+ YOEData Science

Staff Data Scientist owns product measurement strategies, leads cross-team initiatives, sets experimentation standards, and builds scalable analytics for product areas. Requires 10+ years experience, SQL/Python expertise, and strong causal inference skills.

Numeral

Numeral

New York, NY

Software Engineer (AI)
$175k+/yrHybrid7+ YOEML Engineering

Build and operate production AI systems including LLM-powered agents, model orchestration, RAG, and backend APIs for tax automation workflows. Requires 7+ years experience with 2+ in LLMs, backend fluency, and startup intensity.

Cursor

Cursor

San Francisco, CA
Software Engineer, ML Research
No salary listedOn-siteAI Research

Builds distributed training, inference, and data systems for frontier coding models, working with researchers to enable fast iteration. Requires strong infrastructure background and intuitions about language models.

Parallel

Parallel

Palo Alto, CA

Early Career Research Engineer
$150k+/yrOn-siteML Engineering

Designs and trains embedding and retrieval models for AI agents to access web data at hyperscale, balancing research innovation with production efficiency for sub-second latency and fresh indexes.

Parker

Parker

Europe

Data Analyst I
$22k+/yrRemoteData Analytics

Builds analytical metrics, dashboards, automations, and reporting systems for risk and broader business decisions. The role requires strong SQL and Python skills, sound data modeling, business judgment, and the ability to solve ambiguous problems independently.

OpenAI

OpenAI

San Francisco, CA

Software Engineer, Research - Human Data
$131k+/yrHybridData Engineering

Build full-stack systems, tools, and infrastructure for human feedback collection, AI model alignment, and evaluation. Collaborate with researchers to scale production systems and enhance model safety in a fast-paced environment.

Quadric

Quadric

Pune, India

AI Kernel Engineer
No salary listedOn-site5+ YOEAI Research

Develops and optimizes AI and LLM inference kernels for Quadric’s neural processing platform, profiling performance across hardware configurations and improving compiler and runtime components. Requires 5+ years of kernel optimization experience, strong C/C++ and Python skills, and familiarity with CUDA, DSP, NEON, or Triton.

Udio

Udio

New York, NY

Member of Technical Staff - Quantitative Research
$250k+/yrRemote5+ YOEData Science

Pioneers quantitative research for music generation AI using proprietary datasets, designs evaluation frameworks for models, drives data roadmaps, and builds production infrastructure at the intersection of research, engineering, and product. Requires PhD or 5+ years quant experience with strong coding skills.

Luma AI

Luma AI

Palo Alto, CA

Software Engineer, Inference
$188k+/yrHybridML Engineering

Develops and optimizes inference engines for multimodal AI models, integrating new architectures, building scheduling systems, and managing large-scale GPU deployments. Requires strong Python, model serving frameworks like PyTorch/vLLM, and Kubernetes expertise.

Datadog

Datadog

New York, NY
AI Research Scientist – Datadog AI Research (DAIR)
$140k+/yrOn-siteAI Research

Conducts cutting-edge research in Generative AI, building foundation models and autonomous agents for cloud observability, SRE, and code repair. Requires PhD in ML or related field, publications at top conferences, and expertise in PyTorch/TensorFlow distributed training.

Datadog

Datadog

New York, NY

Senior Staff GenAI Engineer - Application Performance Monitoring (APM)
$272k+/yrHybrid10+ YOEML Engineering

Lead GenAI/ML initiatives for Datadog's APM team, designing and deploying models for agentic workflows, automated investigations, and performance optimization. Requires 10+ years experience with 4+ years leading cross-team projects.

Datadog

Datadog

Boston, MA
Staff Software Engineer - ML Observability
$234k+/yrHybridML Engineering

Lead development of LLM observability features at Datadog, building tools for monitoring, tracing, and evaluating AI systems in production. Requires expertise in distributed systems, GenAI applications, and model internals.

Eightsleep

Eightsleep

San Francisco, CA

Machine Learning Engineer (Foundation Models & Personalization)
No salary listedOn-site2+ YOEML Engineering

Builds and deploys ML models for sleep personalization, readiness forecasting, and behavior prediction using foundation models and multimodal data. Requires 2+ years production ML experience with PyTorch/TensorFlow, personalization systems, and strong Python engineering.

Docker

Docker

London, United Kingdom

Principal Software Engineer, Docker Agents
No salary listedRemote10+ YOEAI Research

Leads the technical vision and implementation of Docker’s containerized AI agent platform, including runtime infrastructure, distributed systems, evaluation, and operational excellence. The role requires 10+ years of software engineering experience, principal-level technical leadership, and practical experience with Go, Docker, and LLM-based agent development.

Deepgram

Deepgram

United States

Staff Product Analyst
$180k+/yrRemoteData Analytics

Build analytics foundation for Deepgram's voice AI platform, owning self-serve growth strategy, experiments, and dashboards using SQL, Superset, and Heap. Partner with product team to drive API adoption and inform strategic investments.

Harvey

Harvey

San Francisco, CA

Senior Data Scientist, Marketing
$170k+/yrHybrid6+ YOEData Science

Build marketing data science function, define metrics and dashboards, model channel/campaign performance using stats/ML. Partner with Marketing/GTM/Finance; requires 6+ years experience, SQL/Python/R expertise, modern data stack proficiency.

Deepgram

Deepgram

San Francisco, CA
Research Staff, LLMs
$150k+/yrRemoteAI Research

Conduct cutting-edge research on Large Language Models, focusing on transformer optimization, distributed training, data curation, and RL. Collaborate on experiments, deploy models to production, and drive voice AI innovations.

Deepgram

Deepgram

San Francisco, CA
Research Staff, Voice AI Foundations
$150k+/yrRemoteAI Research

Pioneers Latent Space Models to solve core challenges in voice AI, developing neural audio codecs, generative speech models, and scalable multimodal systems. Requires strong expertise in statistical learning, foundation models, and bridging theory to efficient deployment.

Bazaarvoice

Bazaarvoice

Bengaluru, India

Staff Software Engineer
No salary listedHybrid8+ YOEData Engineering

Leads the architecture and performance of high-throughput real-time streaming and Lakehouse data platforms. The role requires 8+ years of software engineering experience, deep Scala or Java and JVM expertise, extensive Kafka experience, and strong AWS capabilities.

Cerebras Systems

Cerebras Systems

United Arab Emirates

Applied AI/ML Scientist
No salary listedOn-siteAI Research

Develops and customizes large language and deep learning models for customer-specific applications, owning training, fine-tuning, evaluation, and agentic-system development. Requires advanced graduate education, hands-on experience with 1B+ parameter models, Python, PyTorch, and distributed training.

Brex

Brex

San Francisco, CA
Data Engineer
$121k+/yrHybrid3+ YOEData Engineering

Build and maintain scalable data models, pipelines, and Core Data tables to transform raw data into actionable insights. Collaborate with data scientists and business teams using SQL, DBT, Snowflake, Airflow, and Python.

Shield AI

Shield AI

Melbourne, Australia

Computer Vision Engineer
No salary listedOn-siteAI Research

Develops novel C++ computer vision algorithms and real-time perception pipelines for autonomous platforms. The role requires strong computer vision, image processing, machine learning, and high-performance systems expertise, with deep learning and robotics experience advantageous.