Talent.com
TikTok
Large Model Training Acceleration EngineerTikTok • San Jose
No longer accepting applications
Large Model Training Acceleration Engineer

Large Model Training Acceleration Engineer

TikTok • San Jose
30+ days ago
Job type
  • Full-time
Job description

Team Introduction

The Intelligent Creation - AI Platform team is a team focusing on building advanced end-to-end AI production pipelines, including deep learning model training, optimization, deployment and applications. We provide AI capabilities to empower content creation and consumption on TikTok and serve billions of users. We are seeking an experienced AI model optimization engineer with expertise in optimizing AI model training and inference, including distributed training/inference and acceleration. The ideal candidate will work at the cutting edge of AI efficiency, enhancing the performance, scalability, and deployment of large-scale generative AI models. Responsibilities - Optimize large model training pipelines to improve efficiency, speed, and scalability. - Develop and improve distributed training strategies such as data parallelism, model parallelism, pipeline parallelism and communication to accelerate model training. - Benchmark and profile deep learning models to identify performance bottlenecks and optimize computational resources.

Minimum Qualifications: - Master’s or PhD in Computer Science, Electrical Engineering, Artificial Intelligence, or a related field. - 5+ years of experience in AI model training optimization. - Strong software engineering skills, including proficiency in Python, C++, and CUDA. - Strong proficiency in deep learning frameworks such as PyTorch, Megatron and Deepspeed. - Experience with distributed training techniques such as data parallelism, model parallelism, and pipeline parallelism. - Knowledge of transformers and diffusion models.

Create a job alert for this search

Large Model Training Acceleration Engineer • San Jose

Similar jobs

Release Train Engineer (RTE) / Program Manager

Omni InclusiveSanta Clara, CA, United States
Full-time

Release Train Engineer (RTE) / Program Manager.We are seeking a highly accomplished and results-oriented Release Train Engineer (RTE) / Program Manager to lead and drive the successful delivery of ... Show more

 • Promoted

Director of Mergers & Acquisitions - Remote Growth Leader

ETR (Education, Training and Research)Santa Cruz, CA, United States
Remote
Full-time

A leading non-profit organization is seeking a Director of Mergers and Acquisitions to implement its M&A strategy.This remote role requires a Bachelor's degree and minimum 5 years of relevant e... Show more

 • Promoted

Earthmoving Technical Trainer

Peterson Holding CompanySan Leandro, CA, United States
Full-time

Experience Your Purpose At PetersonAt Peterson, we partner with our customers to build the future.For over 85 years, our peoples' work has shaped the communities where we live, where we raise our f... Show more

 • Promoted

MLOps and ML Engineer

APTASKSan Ramon, CA, United States
Full-time

Job TitleThe client is a leading digital transformation consultancy and engineering services company that specializes in driving digital transformation for Fortune 1000 enterprises.The company prov... Show more

 • Promoted

[Remote] Staff Machine Learning Engineer, Closeup Relevance

PinterestPalo Alto, CA, United States
Remote
Full-time

OverviewNote :The job is a remote job and is open to candidates in USA.Pinterest is a platform that inspires creativity and helps users discover new ideas.They are seeking a highly motivated Staff ... Show more

 • Promoted

Engineering Manager - Neural Simulation

Applied IntuitionSunnyvale, CA, United States
Full-time

About Applied IntuitionApplied Intuition, Inc.Founded in 2017 and now valued at $15 billion, the Silicon Valley company is creating the digital infrastructure needed to bring intelligence to every ... Show more

 • Promoted

Foundational ML Researcher -- Live AI, LLM Training, Remote

PathwayPalo Alto, CA, United States
Remote
Full-time

Pathway is seeking R&D Engineers to work on ambitious AI projects involving attention-based models.The role offers a competitive salary with an employee stock option plan and the opportunity to... Show more

 • Promoted

Fire Controlman

U.S. NavyCapitola, CA, United States
Full-time

Job Title :Advanced Electronics / Computer Field (ET/FC) Category / Component :Enlisted Active Overview The Advanced Electronics and Computer Field trains Sailors to maintain, operate, and repair s... Show more

 • Promoted

Software Engineer, Machine Learning

Meta PlatformsNewark, CA, United States
Full-time

Talented Engineers WantedMeta is seeking talented engineers to join our teams in building cutting-edge products that connect billions of people around the world.As a member of our team, you will ha... Show more

 • Promoted

Battery Simulation Engineer

GotionFremont, CA, United States
Full-time

Silicon Valley, CA, currently building a Manufacturing facility in Manteno, IL and has R&D centers in Ohio, China, Japan and Europe.We innovate in the next generation electric vehicle and energ... Show more

 • Promoted

[Remote] Performance Oriented, Senior C++ Engineer for core animation runtime at Rive

Meetingcpp GmbHFremont, CA, United States
Remote
Full-time

Remote Performance Oriented, Sr.C+Engineer for core animation runtime at Rivepublished at 10.Location :RemoteCompany :RiveRemote support for this position :yesRive is a new way to build production-... Show more

 • Promoted

Manager, Sales Training & Operations, TMTT (Remote US)

Edwards LifesciencesSanta Cruz, CA, United States
Remote
Full-time

Patients with mitral and tricuspid heart valve disease often have complex conditions with limited treatment options.Our Transcatheter Mitral and Tricuspid Therapies (TMTT) business unit is boldly p... Show more

 • Promoted

Training

IntersourcesFremont, CA, United States
Full-time

Founded in 2007, InterSources Inc is a Small Business Enterprise (SBE), Minority Business Enterprise (MBE) & Women-Owned Small Business (WOSB) Certified Company specializing in providing IT Consult... Show more

 • Promoted

ENTRY LEVEL AI AND MACHINE LEARNING ENGINEER

SynergisticITFremont, CA, United States
Full-time

Job Opportunity At SynergisticitSynergisticIT wants every candidate to know that the job market is challenging and to stand out, you need to have exceptional skills and technologies and that's wher... Show more

 • Promoted

Electrocardiogram (ECG / EKG) Technician - Entry Level Training Program

Dreambound Inc.Rio del Mar, California, US
Full-time

Note : This is an educational program, not a job.Successful completion of the program does not guarantee employment but will equip you with valuable skills for the healthcare job market.Looking to ... Show more

 • Promoted

Senior Software Engineer - Circuit Simulation - US Remote o

SIEMENS AGFremont, CA, United States
Remote
Full-time

Job Family :SoftwareReq ID :484868Siemens EDA is a global technology leader in Electronic Design Automation software.Our software tools enable companies around the world to develop highly innovativ... Show more

 • Promoted

Principal Machine Learning Engineer

WorkdayPleasanton, CA, United States
Full-time

Your Work Days Are Brighter HereWe're obsessed with making hard work pay off, for our people, our customers, and the world around us.As a Fortune 500 company and a leading AI platform for managing ... Show more

 • Promoted

Lead Principal Product Designer -- UX & Strategy (Remote)

ArtykulaSanta Cruz, CA, United States
Remote
Full-time

Join a forward-thinking company as a Principal Product Designer, where your creativity and strategic thinking will redefine digital experiences for iconic brands.In this dynamic role, you will lead... Show more

 • Promoted

Machine Learning Engineer

Perfict Global, Inc.Fremont, CA, United States
Full-time

Machine Learning EngineerExperience:9years Visa:GC, USC, GCEAD, H4EAD, TN Tax Term:W2 Client:Tesla Location:Fremont, CA, onsiteProject Description:Design, develop and implement critical machine lea... Show more

 • Promoted

Project Engineer

ACCO Engineered SystemsSanta Cruz, CA, United States
Full-time

This position will collaborate closely with a dedicated Project Manager, reporting to the Sales Manager, and be responsible for the technical coordination of assigned projects.This includes fosteri... Show more