Search jobs > San Jose, CA > Internship > Machine learning engineer

Machine Learning Engineer Graduate (On-Device AI) - 2025 Start (PhD)

Littleducklingindy
San Jose, California, US
Internship

TikTok is the premier destination for short-form mobile videos, with a mission to inspire creativity and bring joy. The On-Device AI Team focuses on embedding advanced AI technologies into devices worldwide, enhancing features like visual enhancements and video editing tools.

The company is looking for a talented Machine Learning Engineer Graduate to join their team in 2025.

Scroll down to find the complete details of the job offer, including experience required and associated duties and tasks.

In this role you can expect to have the responsibilities :

  • Develop and enhance the core On-Device AI engine for diverse platforms.
  • Utilize efficient machine learning techniques and hardware optimization to improve algorithmic performance.
  • Integrate state-of-the-art AI algorithms into TikTok and other products.
  • Translate emerging research into practical, user-centric applications.

This role requires you to have :

  • PhD in Computer Science or related field
  • 3+ years of experience in Python / C++
  • Solid CS fundamentals and problem-solving skills
  • 1+ years of experience in AutoML / model optimization
  • Excellent communication and teamwork skills

TikTok promotes an inclusive work environment that celebrates diverse voices. They provide reasonable accommodations for candidates with disabilities, pregnancy, and other protected reasons.

J-18808-Ljbffr

15 hours ago
Related jobs
Promoted
TikTok
San Jose, California

Familiar with at least one framework of TensorFlow / PyTorch / MXNet and its training and deployment details,as well as the training acceleration methods such as mixed precision training and distributed training;. Familiar with common machine/deep learning, causal inference, and operational optimiza...

Promoted
Contextual AI, Inc.
Mountain View, California

We embrace making decisions with insufficient data and focus on being approximately right by relying on anecdotes, intuition, and insights. We are willing to experiment and reimagine how work is done with AI. We do more than listen to our customers, we invent on their behalf and anticipate their nee...

Promoted
TikTok
San Jose, California

Good understanding in one of the following domains: ads bidding & auction, ads quality control, and online advertising systems (familiar with one or more of these terms: CPC/CPM, CTR/CVR, Ranking /Targeting, Conversion/Budget, Campaign/Creative). Our time off and leave plans are: 10 paid holiday...

Promoted
Apple, Inc.
Cupertino, California

We're an applied Machine Learning team that leverages state of the art technologies like generative AI, graph machine learning, and private learning to deliver high quality inferences. As a Machine Learning Engineer on the Relevance and Graph Inference Team, you'll join a phenomenal team of hardwork...

Promoted
TikTok
San Jose, California

Paid Personal Time Off (PPTO) (prorated upon hire and increased by tenure) and 10 paid sick days per year as well as 12 weeks of paid Parental leave and 8 weeks of paid Supplemental Disability. Contribute to optimizing TikTok backend development and operations with advanced engineering methodologies...

Promoted
Qualified Health
Palo Alto, California

Drive the adoption of best practices in machine learning operations (MLOps) to automate model training, deployment, and monitoring. Strong background in MLOps practices including continuous integration, continuous delivery (CI/CD), and automated testing of machine learning models. This role is ideal...

Promoted
Apple
Cupertino, California

The Machine Learning Platform team designs and builds a best of breed platform for building and operating the complex and innovative machine learning systems that deliver highly optimized advertising content to consumers. Are you a results-oriented and versatile engineer who can excel in an Agile en...

ByteDance
San Jose, California

About the team:The Applied Machine Learning (AML) team is committed to the research and deployment of the next-generation of AI core technologies, such as large pretrained models and device-cloud collaboration learning, as well as their wide applications in practice, including but not limited to sea...

Promoted
Karkidi
Cupertino, California

Understanding of machine learning, deep learning, natural language processing, distributed systems, reliability and scalability, containerization, and cloud platforms. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation. In your role as a ...

Promoted
Advanced Micro Devices, Inc.
Santa Clara, California

AMD and its subsidiaries are equal opportunity, inclusive employers and will consider all applicants without regard to age, ancestry, color, marital status, medical condition, mental or physical disability, national origin, race, religion, political and/or third-party affiliation, sex, pregnancy, se...