Machine Learning Engineer Intern (E-Commerce Knowledge Graph) - 2027 Start (PhD)

TikTok; San Jose (CA); On-site; USD 68,000 - 97,000; Part time

Benefits offered by this job

Health insurance

Life insurance

Wellbeing benefits

10 paid holidays per year

Housing allowance

Job summary

TikTok is seeking a Campus Intern to contribute to its Knowledge Graph team in California. You will help build neural network-driven data products for e-commerce, including product tagging, categorization, and category attributes, while collaborating with the recommendation team.

You will work on knowledge graphs, buyer-product connections, and research to explain why similar items outperform others. This role requires pursuing a PhD and strong ML/NLP skills, with exposure to large-scale data

Qualifications

• Currently pursuing a PhD in Software Development, Computer Science, Computer Engineering or related discipline.
• Experience in at least one of: LLMs/NLP, KG, ML, ML frameworks.
• Strong coding skills in Python/Java/C++ and familiarity with DL frameworks.

Responsibilities

• Craft knowledge graphs covering product insights and category/brand connections.
• Architect knowledge maps detailing buyer and product connections.

Skills

LLM/NLP

Natural language generation

Named entity recognition

Machine translation

Question answering

Knowledge graph

Information extraction

Text mining

Text classification

Python

Java

C/C++

Spark

PyTorch

TensorFlow

Keras

Strong analytical skills

Education

PhD in CS/CE/related

Tools

Python

Java

C/C++

Spark

PyTorch

TensorFlow

Keras

Job description

Responsibilities

The Knowledge Graph team is responsible for the neural network of e-commerce, which includes tasks such as categorizing products, influencers, and merchants, mining product information, and providing this data to the recommendation team. Additionally, they manage operations related to product tags and category attributes. When we come across a product, they determine whether it belongs to the clothing or sports category, who thie product will be suitable for, etc. Furthermore, the team conducts competitive research, identifying top-performing items on the platform and analyzing why one similar product outperforms others. They are also involved in brand recognition. This team plays a pivotal role, in bridging various aspects of our business and maintaining close collaboration with other teams.

Responsibilities

The Knowledge Graph team is responsible for the neural network of e-commerce, which includes tasks such as categorizing products, influencers, and merchants, mining product information, and providing this data to the recommendation team. Additionally, they manage operations related to product tags and category attributes. When we come across a product, they determine whether it belongs to the clothing or sports category, who thie product will be suitable for, etc. Furthermore, the team conducts competitive research, identifying top-performing items on the platform and analyzing why one similar product outperforms others. They are also involved in brand recognition. This team plays a pivotal role, in bridging various aspects of our business and maintaining close collaboration with other teams.

Responsibilities:

• Craft intricate knowledge graphs, encompassing areas like product/content insights and category/brand/SPU development;
• Architect knowledge maps detailing buyer and product connections.

Qualifications

Minimum Qualifications

• Currently pursuing a PhD in Software Development, Computer Science, Computer Engineering or a related discipline.
• Experience in at least one of the technological areas:
• LLM, natural language generation, named entity recognition, machine translation, question answering, knowledge graph, information extraction, text mining and classification etc.
• Computer vision, facial recognition, multimodal learning, object detection & tracking, image classification etc.
• Machine learning, transfer learning, graph embedding, representative learning, large-scale optimization, recommendation, reinforcement learning, etc.
• Strong coding skills in at least one of the following programming frameworks, e.g. Python, Java, C/C++, Spark etc.; Familiar with at least one of the following deep learning frameworks, e.g., PyTorch, TensorFlow, Keras etc.
• Strong analytical and problem-solving skills

Preferred Qualifications

• Familiarity with deep learning frameworks like TensorFlow/PyTorch.
• Coursework or projects related to text classification, text matching, sequence labeling, and knowledge graphs.
• Knowledge of domain adaptation, text mining, and unsupervised/semi-supervised methodologies.
• Experience with machine learning and deep learning algorithms, and text representation methods.
• Exposure to large-scale text data processing or tools like Hadoop/Spark/Hive/Flink.
• Work experience in e-commerce or the Internet industry is preferred

Job Information

【For Pay Transparency】Compensation Description (Hourly) - Campus Intern

The hourly rate range for this position in the selected city is $60- $60.

Benefits may vary depending on the nature of employment and the country work location. Interns have day one access to health insurance, life insurance, wellbeing benefits and more. Interns also receive 10 paid holidays per year and paid sick time (56 hours if hired in first half of year, 40 if hired in second half of year). Interns who are not working 100% remote may also be eligible for housing allowance.

The Company reserves the right to modify or change these benefits programs at any time, with or without notice.

For Los Angeles County (unincorporated) Candidates

• Interacting and occasionally having unsupervised contact with internal/external clients and/or colleagues;
• Appropriately handling and managing confidential information including proprietary and trade secret information and access to information technology systems; and
• Exercising sound judgment.

About TikTok

TikTok is the leading destination for short-form mobile video. At TikTok, our mission is to inspire creativity and bring joy. TikTok's global headquarters are in Los Angeles and Singapore, and we also have offices in New York City, London, Dublin, Paris, Berlin, Dubai, Jakarta, Seoul, and Tokyo.

Why Join Us

Inspiring creativity is at the core of TikTok's mission. Our innovative product is built to help people authentically express themselves, discover and connect - and our global, diverse teams make that possible. Together, we create value for our communities, inspire creativity and bring joy - a mission we work towards every day.

We strive to do great things with great people. We lead with curiosity, humility, and a desire to make impact in a rapidly growing tech company. Every challenge is an opportunity to learn and innovate as one team. We're resilient and embrace challenges as they come. By constantly iterating and fostering an "Always Day 1" mindset, we achieve meaningful breakthroughs for ourselves, our company, and our users. When we create and grow together, the possibilities are limitless. Join us.

Diversity & Inclusion

TikTok is committed to creating an inclusive space where employees are valued for their skills, experiences, and unique perspectives. Our platform connects people from across the globe and so does our workplace. At TikTok, our mission is to inspire creativity and bring joy. To achieve that goal, we are committed to celebrating our diverse voices and to creating an environment that reflects the many communities we reach. We are passionate about this and hope you are too.

TikTok Accommodation

TikTok is committed to providing reasonable accommodations in our recruitment processes for candidates with disabilities, pregnancy, sincerely held religious beliefs or other reasons protected by applicable laws. If you need assistance or a reasonable accommodation, please reach out to us at https://tinyurl.com/RA-request