Skip to content
CriblCriblUnited States

Sr Machine Learning Engineer, AI Research

Design, train, and productionize ML models and pipelines for AI-enabled security/observability platforms. Requires 4+ years experience with Python, PyTorch/TensorFlow, and modern ML techniques.

185k – 215k
Remote4+ YOEML Engineering

About the role

Responsibilities

  • Design, train, and evaluate machine learning models across research and applied AI initiatives
  • Run rapid, iterative experiments to test hypotheses and surface insights that drive model improvements
  • Collaborate closely with researchers and engineers to translate cutting-edge academic advances into practical, production-ready systems
  • Build and maintain robust ML pipelines for data ingestion, feature engineering, model training, and evaluation
  • Optimize model performance through fine-tuning, hyperparameter search, and architecture experimentation
  • Contribute to a culture of rigorous experimentation; track results, document findings, and share learnings with the broader team
  • Stay current with the latest developments in ML and AI research, and proactively identify opportunities to apply them

Requirements

  • Bachelor's degree in Computer Science, Mathematics, Statistics, or a related field with 4+ years of industry or research experience (Master's or PhD a plus)
  • Deep hands-on experience training and evaluating ML models, including language models
  • Strong proficiency in Python and ML frameworks such as PyTorch or TensorFlow
  • Familiarity with MLOps tooling and infrastructure (e.g., MLflow, Weights & Biases, Kubeflow, or similar)
  • Solid understanding of modern NLP, computer vision, and/or reinforcement learning techniques
  • Strong ability to move fast without sacrificing rigor; know when to prototype and when to productionize
  • Excellent communication skills with the ability to clearly present experimental results to both technical and non-technical stakeholders

Nice-to-Haves

  • Master's or PhD degree

Skills

PythonPyTorchTensorFlowMLflowWeights & BiasesKubeflowNLPComputer VisionReinforcement LearningMLOps

Similar roles

ML Engineering jobs
Cribl

Sr Software Engineer, Cribl AI

CriblUnited States

Build and productionize AI integrations across Cribl's observability platform. Requires 5+ years building AI features end-to-end with fullstack experience in React, TypeScript/JavaScript, and backend systems plus hands-on LLM/agent work.

185k – 215k
Remote5+ YOEML Engineering
Coinbase

Senior Software Engineer - AI Platform Team

CoinbaseUnited States

Senior Software Engineer building and operating Coinbase's central LLM and AI agent platform infrastructure. Owns LLM Gateway, knowledge bases, agent runtime, FinOps, evaluation tooling, and cross-team applied AI strategy. Requires 5+ years building ML/AI infrastructure at scale.

186k – 219k
Remote5+ YOEML Engineering
Datadog

Senior AI Engineer - APM Experiences

DatadogNew York, NY

Leads development of LLM- and agent-based features for Datadog's APM to detect, resolve, and prevent performance issues using traces, metrics, and logs. Requires 4+ years in backend/ML systems, production AI experience, and distributed systems expertise.

187k – 240k
On-site4+ YOEML Engineering
Datadog

Senior Software Engineer - AI Code Gen - Agent Engineer

DatadogNew York, NY

Build AI-assisted tools using LLMs to generate tests, fix bugs, and boost developer productivity at Datadog. Requires 5+ years backend experience in Go/Java/Python and shipping LLM software to production.

187k – 240k
Hybrid5+ YOEML Engineering
Zoox

Senior Machine Learning Engineer - Scene Understanding

ZooxFoster City, CA +1

Develops advanced Vision-Language-Action models for robotaxi scene understanding, detecting hazards and enabling safe driving. Leads data strategies, post-training of large models, and deployment using PyTorch and production ML pipelines. Requires MS/PhD in CS and deep learning expertise.

189k – 290k
HybridML Engineering