Cloudflare

Member of Technical Staff - Workers AI Applied Research

Hybrid. Posted 4 months ago

The role

You will define how machine learning models run across Cloudflare's global network. This role involves applied ML, inference optimization, evaluation, and production engineering to bring models into production.

What you will do

  • Develop, optimize, and productionize machine learning models for Cloudflare’s serverless inference platform.
  • Build benchmarking and evaluation frameworks to measure latency, throughput, cost efficiency, and model behavior.
  • Improve inference performance through quantization, batching, caching, model compilation, runtime tuning, and accelerator-aware optimization.
  • Partner with systems engineers to integrate models into distributed inference infrastructure.
  • Drive improvements to model deployment workflows, including validation, rollout safety, observability, regression testing, and operational readiness.
  • Collaborate with product and engineering teams to translate customer requirements into scalable ML capabilities.
  • Mentor engineers, contribute to technical direction, and raise the quality bar for production ML engineering practices.

Required qualifications

  • Experience building, optimizing, and operating machine learning models in production environments.
  • Strong proficiency with Python and modern ML frameworks such as PyTorch, TensorFlow, JAX, or equivalent.
  • Hands-on experience with inference optimization techniques for large-scale models, including quantization, batching, caching, compilation, and serving runtime tuning.
  • Experience with large-scale inference serving frameworks or runtimes such as SGLang, vLLM, TensorRT-LLM, ONNX Runtime, Triton, llama.cpp, or similar.
  • Familiarity with LLMs, speech models, vision models, embeddings, multimodal models, retrieval-augmented generation, or other modern deep learning architectures.
  • Experience optimizing models for GPUs or specialized accelerators.
  • Strong understanding of production ML concerns, including evaluation, monitoring, model regressions, rollout safety, and reliability.
  • Ability to work across ML and systems boundaries, including familiarity with distributed systems, networking, or serverless platforms.
  • Track record of leading complex technical projects and mentoring other engineers.

Nice to have

  • Experience contributing to open source ML tooling, model serving frameworks, or inference runtimes.

Benefits

  • Medical/Rx Insurance
  • Dental Insurance
  • Vision Insurance
  • Flexible Spending Accounts
  • Commuter Spending Accounts
  • Fertility & Family Forming Benefits
  • On-demand mental health support and Employee Assistance Program
  • Global Travel Medical Insurance
  • Short and Long Term Disability Insurance
  • Life & Accident Insurance
  • 401(k) Retirement Savings Plan
  • Employee Stock Participation Plan

Before you apply

  • Applicants who progress to the offer stage may be asked to attend an in-person interview within one of the Cloudflare Offices or Cloudflare Hubs.
  • Offer of employment may be conditioned on authorization to receive software or technology controlled under U.S. export laws without sponsorship for an export license.

About the company

Cloudflare runs one of the world’s largest networks that powers millions of websites and other Internet properties. The company protects and accelerates any Internet application online without adding hardware, installing software, or changing a line of code.

Member of Technical Staff - Workers AI Applied Research at Cloudflare | JobZoey