Logo of Huzzle

Machine Learning Engineer Intern (TikTok - PGC - LLM/Content Understanding) - 2025 Start (PhD)

image

TikTok

2mo ago

  • Internship
    Full-time
    Summer Internship
  • Data
    IT & Cybersecurity
  • San Jose

AI generated summary

  • You should have a Master’s or PhD in a relevant field, experience with LLMs, Python, deep learning frameworks, and scalable ML pipelines, plus problem-solving skills and collaboration ability.
  • You will develop and optimize LLM models, design AI architectures, implement innovative techniques, collaborate on AI solutions, and enhance training pipelines for improved performance.

Requirements

  • Minimum Qualifications:
  • Master’s or PhD in Computer Science, Machine Learning, AI, or a related field.
  • Strong experience in training, fine-tuning, and deploying LLMs (e.g., GPT, LLaMA, PaLM, Mistral).
  • Proficiency in Python and deep learning frameworks (e.g., PyTorch, TensorFlow, JAX).
  • Solid understanding of transformers, retrieval-augmented generation (RAG), and prompt engineering.
  • Hands-on experience with scalable machine learning pipelines and working with distributed computing frameworks (e.g., Ray, Spark).
  • Strong problem-solving skills and the ability to work in cross-functional teams.
  • Preferred Qualifications:
  • Experience in LLM inference optimization, distillation, or model compression.
  • Knowledge of vector databases (e.g., FAISS, Milvus), embedding search, and retrieval systems.
  • Familiarity with real-world deployment of LLMs in production environments at scale.
  • Contributions to research papers, open-source AI projects, or AI conferences.

Responsibilities

  • Develop and optimize LLM-based content understanding models to enhance video, text, and multimodal content analysis.
  • Build and fine-tune domain-specific LLMs for creator-related applications.
  • Design scalable, high-performance AI architectures, working closely with Applied AI infra engineers to ensure efficient model serving.
  • Implement innovative techniques such as Reinforcement Learning from Human Feedback (RLHF), Retrieval-Augmented Generation (RAG), and multi-modal learning to improve AI-driven insights.
  • Collaborate with product teams to translate business challenges into AI solutions and drive measurable product impact.
  • Optimize training, inference, and fine-tuning pipelines to improve efficiency, scalability, and real-time performance.
  • Stay at the cutting edge of LLM research, exploring and applying the latest advancements to YouTube’s ecosystem.

FAQs

What is the duration of the internship?

The internship program runs for 12 weeks, beginning in May/June 2025.

What are the available start dates for the internship?

The available start dates for the internship are May 12, May 19, May 27, June 9, and June 23.

What educational qualifications are required for this position?

Candidates must have a Master’s or PhD in Computer Science, Machine Learning, AI, or a related field.

What kind of experience is preferred for this internship?

Preferred candidates will have experience in LLM inference optimization, model compression, and familiarity with real-world deployment of LLMs in production environments at scale.

Are there any specific programming skills required?

Yes, proficiency in Python and deep learning frameworks such as PyTorch, TensorFlow, or JAX is required.

What kind of projects will the intern work on?

The intern will work on developing and optimizing LLM-based content understanding models, building domain-specific LLMs, and collaborating with product teams to translate business challenges into AI solutions.

Will there be opportunities for personal and professional development during the internship?

Yes, interns will have access to social events and enriching development workshops for personal and professional growth.

What is the hourly rate for this position?

The hourly rate range for this position is $45 to $60.

Does TikTok provide any benefits for interns?

Yes, interns have day one access to health insurance, life insurance, wellbeing benefits, 10 paid holidays per year, and paid sick time.

Is there a limit on the number of positions a candidate can apply for?

Yes, candidates can apply to a maximum of two positions.

Does TikTok have a commitment to diversity and inclusion?

Yes, TikTok is committed to creating an inclusive space and values diverse voices, skills, and experiences in the workplace.

Entertainment & Media
Industry
10,001+
Employees
2017
Founded Year

Mission & Purpose

TikTok is a social media platform where users create and share short videos on various topics, including entertainment, education, and lifestyle. Their ultimate goal is to inspire creativity and bring joy to people's lives through engaging and entertaining content. TikTok's purpose is to provide a platform for self-expression and connection, fostering a global community where individuals can express themselves authentically and connect with others who share similar interests and passions.