AI Infra Engineer Graduate (Recommendation & LLM) 2027 Start (PhD)

<strong>Responsibilities<br><br></strong>About the Team<br><br>We are looking for a passionate Software Engineer to join our Model Infrastructure team, building the next generation of infrastructure for TikTok's For You recommendation system and Large Language Models (LLMs).<br><br>Our team develops the core training and serving infrastructure that powers one of the world's largest recommendation systems, enabling billions of personalized recommendations every day. We are also advancing the next generation of AI infrastructure for foundation models and LLMs, driving innovation in large-scale model training, online inference, and GPU optimization.<br><br>As part of the team, you will work on distributed training and inference systems, high-performance GPU computing, and scalable LLM infrastructure. You'll collaborate closely with experienced engineers and researchers to transform cutting-edge AI technologies into production systems that directly impact the experience of hundreds of millions of TikTok users. This role is ideal for candidates who are passionate about LLM systems, distributed computing, GPU programming, and building AI systems at massive scale.<br><br>We are looking for talented individuals to join our team. As a graduate, you will get opportunities to pursue bold ideas, tackle complex challenges, and unlock limitless growth.<br><br>Successful candidates must be able to commit to an onboarding date by the end of the year. Please state your availability and graduation date clearly in your resume.<br><br><strong>Responsibilities<br><br></strong><ul><li> Build and optimize infrastructure for large-scale model training and online inference.</li><li> Develop distributed systems supporting large recommendation models and LLMs.</li><li> Improve training and inference performance through GPU optimization and efficient communication.</li><li> Collaborate with researchers to develop and deploy LLM training and serving solutions.</li><li> Analyze system bottlenecks and implement performance optimizations.<br><br></li></ul><strong>Qualifications<br><br></strong>Minimum Qualifications<br><br><ul><li> Individuals who are completing or have recently completed a PhD degree in Computer Science or a related discipline.</li><li> Strong programming skills in C++ or Python.</li><li> Good understanding of data structures, algorithms, and computer systems.</li><li> Familiarity with PyTorch or TensorFlow.</li><li> Knowledge of Transformer architectures and Large Language Models (LLMs).</li><li> Strong problem-solving skills and a passion for building large-scale AI systems.<br><br></li></ul>Preferred Qualifications<br><br><ul><li> Hands-on experience with LLM training or inference through research, internships, or open-source projects.</li><li> Familiarity with distributed training concepts (e.g., DP, TP, PP, FSDP, ZeRO).</li><li> Experience with GPU programming using CUDA, Triton, or similar technologies.</li><li> Understanding of LLM serving techniques such as KV Cache, Continuous Batching, or FlashAttention.</li><li> Contributions to open-source projects or research in machine learning systems, distributed systems, or LLM infrastructure.<br><br></li></ul>Job Information<br><br><strong>【For Pay Transparency】Compensation Description (Annually)<br><br></strong>The base salary range for this position in the selected city is $162000 - $387600 annually.<br><br>Compensation may vary outside of this range depending on a number of factors, including a candidate’s qualifications, skills, competencies and experience, and location. Base pay is one part of the Total Package that is provided to compensate and recognize employees for their work, and this role may be eligible for additional discretionary bonuses/incentives, and restricted stock units.<br><br>Benefits may vary depending on the nature of employment and the country work location. Employees have day one access to medical, dental, and vision insurance, a 401(k) savings plan with company match, paid parental leave, short-term and long-term disability coverage, life insurance, wellbeing benefits, among others. Employees also receive 10 paid holidays per year, 10 paid sick days per year and 17 days of Paid Personal Time (prorated upon hire with increasing accruals by tenure).<br><br>The Company reserves the right to modify or change these benefits programs at any time, with or without notice.<br><br><strong>For Los Angeles County (unincorporated) Candidates:<br><br></strong>Qualified applicants with arrest or conviction records will be considered for employment in accordance with all federal, state, and local laws including the Los Angeles County Fair Chance Ordinance for Employers and the California Fair Chance Act. Our company believes that criminal history may have a direct, adverse and negative relationship on the following job duties, potentially resulting in the withdrawal of the conditional offer of employment:<br><br><ul><li> Interacting and occasionally having unsupervised contact with internal/external clients and/or colleagues;</li><li> Appropriately handling and managing confidential information including proprietary and trade secret information and access to information technology systems; and</li><li> Exercising sound judgment.<br><br></li></ul><strong>About TikTok<br><br></strong>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.<br><br>Why Join Us<br><br>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.<br><br>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.<br><br>Diversity & Inclusion<br><br>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.<br><br>TikTok Accommodation<br><br>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

Back to blog

Other Jobs To Apply

Let's begin! Director Product Manager- Banking (15734)

Account Representative- Personal Lines

Analyst, Bridge Services, Revenue Management-Global (Remote)

Workday Configuration SME - Learning & Recruiting :: Remote :: W2 Position

Finance Operator

(Remote) Technical Consultant – Oracle DBA

Assistant Manager, WW New Media Servicing

Sr. Sales Director – Mobile & Tech

Account Executive, Mid-Market (Remote)

Ramp Service Agent (up to $6,000 hiring bonus)

Global Alliance Lead, Capgemini

REMOTE- Senior Platform Software Engineer- Oracle Health

Customer Service Representative- REMOTE

Marketing Technologist (Remote) - Full-time

Patient Access Representative - Remote

Brand Marketing Manager

Occupational Therapist, Virtual

Senior Product Adoption Strategist (Shopping)

Sr. Tech Business Developer, Just Walk Out Technology by Amazon - AWS Solutions

Data Product Manager (Remote)

Hybrid Medical Scribe - Primary Care, AZ (Remote 2x/wk) Job at Quest Oracle Community in Tucson

Customer Service Representative - Part-Time

Permanent Job Hobby Lobby Remote Jobs Work From Home - Part Time $26/Hour - MySmartPros Job ID- 757

Personal Banker Greater New Haven Area

Data Reporting and Analytics Consultant II, Programming

Staff Software Engineer, Front End

Remote Accounts Payable Specialist – Client-Focused

Litigation Insurance Adjuster (Remote USA)

Remote Legal Counsel — SaaS, Data Privacy & AI

Remote Principal Full Stack Software Engineer

Senior Customer Success Architect

QA Lead / Senior Quality Assurance Engineer

Remote E-Discovery Application Administrator II

Clinical Review Pharmacist (Fully Remote)

Remote Help Desk Manager — AI & Service Excellence

National Account Lead, Walmart & Costco (Remote)

Remote Business Process Improvement Manager

Financial Services Professional – Dental Professionals Encouraged to Apply

VP, Data & Insights

Senior Director of Marketing (Remote)

Data Entry Clerk Work From Home - Part Time Focus Group Panelists (Up To $750/Week)

Remote Inside Sales Representative New Acquisition

Sales Representative (Remote) - Now Hiring

Junior Sales Associate - Work From Anywhere - Now Hiring

Adjunct Part Time Online 16-week general education courses in English, Speech, Math and Anatomy

Coca-Cola Data Entry Clerk (Remote)

Entry‑Level Remote Customer Support Specialist – Live Chat, No Phone, Flexible Hours, $35/hr, Global Opportunities

Underwriting Consultant at Pacific Life: 100% Remote (AZ)

Work At Home Contractor

Nurse Practitioner — Telehealth Chronic Care Management (Full-time or Part-time, Remote)

Common Interview Questions And Answers

1. HOW DO YOU PLAN YOUR DAY?

This is what this question poses: When do you focus and start working seriously? What are the hours you work optimally? Are you a night owl? A morning bird? Remote teams can be made up of people working on different shifts and around the world, so you won't necessarily be stuck in the 9-5 schedule if it's not for you...

2. HOW DO YOU USE THE DIFFERENT COMMUNICATION TOOLS IN DIFFERENT SITUATIONS?

When you're working on a remote team, there's no way to chat in the hallway between meetings or catch up on the latest project during an office carpool. Therefore, virtual communication will be absolutely essential to get your work done...

3. WHAT IS "WORKING REMOTE" REALLY FOR YOU?

Many people want to work remotely because of the flexibility it allows. You can work anywhere and at any time of the day...

4. WHAT DO YOU NEED IN YOUR PHYSICAL WORKSPACE TO SUCCEED IN YOUR WORK?

With this question, companies are looking to see what equipment they may need to provide you with and to verify how aware you are of what remote working could mean for you physically and logistically...

5. HOW DO YOU PROCESS INFORMATION?

Several years ago, I was working in a team to plan a big event. My supervisor made us all work as a team before the big day. One of our activities has been to find out how each of us processes information...

6. HOW DO YOU MANAGE THE CALENDAR AND THE PROGRAM? WHICH APPLICATIONS / SYSTEM DO YOU USE?

Or you may receive even more specific questions, such as: What's on your calendar? Do you plan blocks of time to do certain types of work? Do you have an open calendar that everyone can see?...

7. HOW DO YOU ORGANIZE FILES, LINKS, AND TABS ON YOUR COMPUTER?

Just like your schedule, how you track files and other information is very important. After all, everything is digital!...

8. HOW TO PRIORITIZE WORK?

The day I watched Marie Forleo's film separating the important from the urgent, my life changed. Not all remote jobs start fast, but most of them are...

9. HOW DO YOU PREPARE FOR A MEETING AND PREPARE A MEETING? WHAT DO YOU SEE HAPPENING DURING THE MEETING?

Just as communication is essential when working remotely, so is organization. Because you won't have those opportunities in the elevator or a casual conversation in the lunchroom, you should take advantage of the little time you have in a video or phone conference...

10. HOW DO YOU USE TECHNOLOGY ON A DAILY BASIS, IN YOUR WORK AND FOR YOUR PLEASURE?

This is a great question because it shows your comfort level with technology, which is very important for a remote worker because you will be working with technology over time...