Research Scientist / AI Software Engineer, LLM/Agent Platform - TikTok - Content Ecology AI Innovation & Platform - TikTok Content Ecology

What you'll do

• Build reusable pipelines covering model and agent data processing, training, deployment, monitoring, and governance.
• Develop platform infrastructure, observability, evaluation, and high-throughput retrieval/RAG systems in collaboration with algorithm teams.
• Design and optimize agentic AI components, including planning, tool use, and memory.

What they ask for

• BS/BA or Master's in Computer Science or a related technical field, or equivalent technical experience.
• 2+ years of hands-on software engineering experience focused on machine learning, distributed systems, or AI infrastructure.
• Strong proficiency integrating AI tools into knowledge discovery and research workflows; familiarity with evaluation frameworks and experimental reproducibility.
• Expertise in deep learning frameworks and tensor libraries such as PyTorch, TensorFlow, or JAX/FLAX; solid understanding of machine learning fundamentals and the modern AI stack.

Nice to have

• PhD in Computer Science or a related technical discipline; 5+ years as an architect or in a technical leadership position.
• Experience with ML infrastructure, large-scale model training and inference, agent architectures, or publications in systems or machine learning conferences.

Tools

PyTorch, TensorFlow, JAX/FLAX, Triton, TensorRT-LLM, FAISS, Milvus, DeepSpeed/ZeRO

AI focus

Builds LLM/agent infrastructure and agentic AI systems, including model lifecycle pipelines, evaluation, retrieval/RAG, planning, tool use, and memory.