T

TikTok

Short-form video platform with ads

Senior Research Scientist/AI Software Engineer - Multiple Teams

AI Application EngineeringConfirmed live in the last 24 hours
Full-timeSan Jose, CA, USAIn-person

About the Job

Requirements

  • A Bachelor of Science, Bachelor of Arts, or Master's degree in Computer Science or a related technical field, or equivalent technical experience.
  • At least 5 years of hands-on software engineering experience focused on machine learning, distributed systems, or AI infrastructure.
  • Proficiency integrating AI tools into knowledge discovery and research workflows.
  • Familiarity with building robust evaluation frameworks and ensuring experimental reproducibility.
  • Expertise in deep learning frameworks and tensor libraries such as PyTorch, TensorFlow, and JAX/FLAX.
  • A solid understanding of machine learning fundamentals and the modern AI stack.
  • Excellent communication skills to collaborate across teams.

Responsibilities

  • Architect and build standardized, configurable, and reusable pipelines for the entire lifecycle of models and agents, from data processing and training through deployment, monitoring, and governance.
  • Partner with algorithm teams to understand their needs and provide an infrastructure platform that accelerates their research and development cycles.
  • Build observability and evaluation frameworks to ensure the reproducibility, reliability, and cost-efficiency of AI workloads at scale.
  • Design and implement core platform infrastructure, including model and agent registries, feature stores, and high-throughput retrieval and retrieval-augmented generation systems.
  • Design, build, and optimize advanced agentic artificial intelligence systems, focusing on planning, tool use, and memory.

Desired Qualifications

  • A PhD in Computer Science or a related technical discipline.
  • At least 5 years of experience as an architect or in a technical leadership position.
  • Experience with the machine-learning infrastructure ecosystem, including GPU scheduling, model serving with Triton and TensorRT-LLM, vector databases such as FAISS and Milvus, and MLOps principles.
  • Experience with large-scale model training and inference, including distributed training, KV-cache-aware serving, GPU/accelerator optimization, and high-performance networking such as RDMA and NCCL.
  • Experience optimizing the performance of large-model training and inference, including DeepSpeed/ZeRO and vLLM.
  • Deep knowledge of agent architectures, including planning, tool use with LangChain and LlamaIndex, and memory systems.
  • Publications in systems and/or machine-learning conferences such as NeurIPS, OSDI, SOSP, ASPLOS, or MLSys.

About the Company

TikTok

Short-form video platform with ads

TikTok is a short-form mobile video platform that lets people create, discover, and share quick videos. Users upload vertical clips, edit them with built-in tools, and browse a personalized feed powered by an algorithm that highlights relevant content. Advertisers can use in-feed ads, branded hashtags, and the Business Center to plan and measure campaigns, making it easy for brands to reach a global audience. Its goal is to inspire creativity and bring joy by helping people express themselves and by connecting brands with a wide audience.

Company size
10,001+
Company stage
Grant
Total funding
$740K
Headquarters
Santa Monica, CA
Founded
2016

Recently Posted Jobs

Explore more opportunities at TikTok

See more jobs