Skip to content
ClickUpClickUpUnited States

Staff AI Engineer - AI Platform

Builds scalable AI platform services for deploying and managing LLMs, integrates with providers like OpenAI and Anthropic, and applies AI to enhance product features. Requires deep backend expertise, distributed systems, and MLOps experience.

250k – 300k/yr
RemoteML Engineering

About the role

Key Responsibilities

  • Architect, design, and implement scalable AI platform services that support the deployment, orchestration, and lifecycle management of LLMs and other AI models.
  • Apply LLMs and other AI technologies directly to build and enhance ClickUp’s intelligent features, working closely with product and engineering teams to deliver impactful solutions.
  • Build and maintain robust APIs and backend systems that enable seamless integration of AI-powered features into ClickUp’s core platform.
  • Develop infrastructure for model serving, monitoring, logging, and automated evaluation to ensure high reliability and performance of AI services in production.
  • Integrate with multiple LLM providers (e.g., OpenAI, Anthropic, Google) and manage model selection, routing, and fallback strategies for optimal performance and cost.
  • Drive the adoption of best practices in AI privacy, security, and compliance, including data anonymization, secure data handling, and regulatory adherence.
  • Optimize platform performance, scalability, and cost-efficiency, leveraging cloud-native technologies and distributed systems.
  • Stay current with advancements in AI infrastructure, MLOps, and LLM applications, and proactively incorporate relevant innovations into ClickUp’s AI platform.
  • Collaborate cross-functionally with product, frontend, and data teams to deliver seamless, reliable, and user-centric AI experiences.

Qualifications

  • Extensive experience designing and building scalable AI/ML platforms or infrastructure in a production environment.
  • Proven track record of applying LLMs and AI models to real-world product features and user-facing solutions.
  • Deep expertise in backend engineering, distributed systems, and cloud-native technologies (e.g., Kubernetes, Docker, AWS/GCP/Azure).
  • Proven experience integrating and managing multiple LLMs and AI models, with a strong understanding of their operational requirements and limitations.
  • Proficiency in orchestration frameworks and workflow engines (e.g., LangGraph, Airflow, Kubeflow, Ray).
  • Strong programming skills in Python, Go, TypeScript or similar languages used for backend and AI platform development.
  • Experience with MLOps best practices, including model deployment, monitoring, logging, and automated evaluation.
  • Demonstrated ability to address AI privacy and security challenges, including data anonymization and compliance with data protection regulations.
  • Familiarity with search technologies and their integration into AI-driven applications.

Skills

LLMsPythonGoTypeScriptKubernetesDockerAWSGCPAzureMLOpsLangGraphAirflowKubeflowRayOpenAI

Similar roles

ML Engineering jobs
Axion

Staff Software Engineer, Agentic Platform

AxionSan Francisco, CA +1

Senior individual contributor architecting and scaling agentic LLM systems that turn messy manufacturing data into reliable root-cause insights. Owns orchestration, retrieval, evaluation, and guardrails for non-deterministic production systems.

250k – 270k/yr
Hybrid7+ YOEML Engineering
The Voleon Group

Member of Research Staff, Optimization

The Voleon GroupBerkeley, CA +1

Conduct optimization research and implement large-scale constrained optimization models that drive real-time trading decisions, working across the full research lifecycle from theory to production. Requires PhD-level coursework and strong applied research background in optimization.

250k – 275k/yr
Hybrid7+ YOEML Engineering
Nuance Labs

Member of Technical Staff — RL Research

Nuance LabsSeattle, WA

New/recent PhD to own RL and post-training for large-scale omni models. Build and scale the full RL/post-training stack including rollout, optimization, reward modeling, and evaluation for real-time audiovisual AI.

250k – 350k/yr
On-siteEntry levelML Engineering
Labelbox

Staff Software Engineer, AI Data Platform

LabelboxSan Francisco, CA

Staff-level engineer building AI data platform infrastructure, eval systems, and agent-first tooling for frontier labs. Requires 4+ years shipping production systems, full-stack experience, and deep TypeScript/Python proficiency.

250k – 280k/yr
Hybrid4+ YOEML Engineering
Nuance Labs

Member of Technical Staff — Model Optimization and Inference

Nuance LabsSeattle, WA

Optimize inference for real-time multimodal AI avatars. Specialize in LLM and diffusion model serving, KV cache strategies, quantization, and low-latency frameworks like vLLM and TensorRT-LLM.

250k – 350k/yr
On-site7+ YOEML Engineering