Senior LLM & Agent Platform Scientist

TikTok; San Jose (CA); On-site; USD 254,000 - 588,000; Full time

Benefits offered by this job

Medical insurance

Dental insurance

Vision insurance

401(k) with match

Parental leave

Disability insurance

Life insurance

Wellbeing benefits

Paid holidays

Paid sick days

Paid personal time

Job summary

TikTok is seeking an experienced AI Infrastructure Engineer to join the Content Ecology Algorithm Team in California. You will architect reusable pipelines spanning data processing, training, deployment, and governance, collaborating with algorithm teams to accelerate research and development.

The role focuses on observability, scalability, and cost-efficiency of AI workloads at scale. Candidates should bring 5+ years in software engineering with ML infrastructure, proficiency in PyTorch,

Qualifications

• BS/BA or MS in CS or equivalent technical experience.
• 5+ years in software engineering with ML/AI infra.
• Strong proficiency in AI tools for research workflows.
• Familiar with evaluation frameworks and reproducibility.
• Expertise in PyTorch, TensorFlow, JAX/FLAX.
• Understanding of the modern AI stack.
• Excellent communication across teams.

Responsibilities

• Architect and build standardized, configurable, reusable pipelines for ML models and agents lifecycle.
• Partner with algorithm teams to provide infrastructure that accelerates research.
• Build observability and evaluation frameworks for reproducibility and cost-efficiency.
• Design core platform infra, including registries and feature stores.
• Design and optimize Agentic AI systems focusing on planning, tool use, memory.

Skills

CS degree or equivalent

5+ years software engineering

ML infra experience

AI tools in workflows

Evaluation frameworks

Deep learning frameworks

Communication across teams

Education

BS/BA or MS in CS

Tools

PyTorch

TensorFlow

JAX/FLAX

Triton

TensorRT-LLM

FAISS

Milvus

DeepSpeed

Job description

TikTok is seeking an experienced AI Infrastructure Engineer to join the Content Ecology Algorithm Team in California. You will architect reusable pipelines spanning data processing, training, deployment, and governance, collaborating with algorithm teams to accelerate research and development.

The role focuses on observability, scalability, and cost-efficiency of AI workloads at scale. Candidates should bring 5+ years in software engineering with ML infrastructure, proficiency in PyTorch,