Distributed Training

Distributed Training Developer Jobs

2 jobs found

3d ago

Machine Learning Engineer, Commerce Ads Ranking

TikTok San Jose, CA

San Jose, CA

3d ago

$156k - $317k/yr Full time Ad Tech Ai Ml Distributed Training

Apply Now

Job Description

TikTok seeks a Machine Learning Engineer for Commerce Ads Ranking in San Jose, CA on-site, with a yearly salary range of USD 156,000 - 316,800.

Responsibilities

Enhance ads delivery models across recall to ranking, covering feature engineering, model architecture, automated feature crossing, cold-start handling, delayed feedback modeling, multi-task learning, and sequence modeling
Co design algorithms and system components for retrieval, sample mining, and processing of long sequences
Investigate and build scalable distributed training pipelines, including GPU tuning, feature processing, and synchronous training, with privacy-preserving considerations to boost delivery efficiency
Develop next generation recommendation systems within an LLM driven framework, including entity understanding, end-to end Generative Recommendation, sequence only recommendations, and Mixture of Export

Requirements

Strong coding skills in C++ and Python, with solid data structures and algorithms fundamentals
Solid foundation in machine learning and familiarity with common models such as FM, LR, GBDT, DNN, and Wide & Deep
Excellent learning agility, collaborative mindset, and strong analytical and problem solving abilities; publications in KDD, NeurIPS, SIGIR, or related ML/data mining competitions are a plus
Bachelor’s degree or higher in computer science or a related field

Technologies

C++, Python
LLM

Benefits

Medical, dental, and vision insurance from day one
401(k) plan with company match
Paid parental leave
Short-term and long-term disability coverage
Life insurance and wellbeing benefits
10 paid holidays per year
10 paid sick days per year
17 days of Paid Personal Time, prorated at hire with tenure-based accruals
Eligible for discretionary bonuses or incentives
Restricted stock units

Los Angeles County Candidates

Interacting with internal or external clients and colleagues, occasionally with unsupervised contact
Managing confidential information including proprietary data and access to IT systems
Exercising sound judgment in decision making

View full job page →

23d ago

Software Engineer- AI/ML, AWS Neuron Distributed Training - Performance Optimization

Annapurna Labs (U.S.) Inc. Cupertino, CA

Cupertino, CA

23d ago

$144k - $194k/yr Full time AWS Aws Neuron Cloud

Apply Now

Job Description

Annapurna Labs (U.S.) Inc., based in Cupertino, is seeking a Software Engineer to join the AWS Neuron Distributed Training team. This role centers on performance optimization for distributed training on Trainium, collaborating with PyTorch, JAX, and the Neuron compiler and runtime to boost throughput and efficiency for large-scale models.

Responsibilities

Lead efforts to optimize distributed training performance on Trainium, with a primary focus on maximizing training throughput, model FLOPs utilization, and overall efficiency across the Neuron software stack.
Collaborate across PyTorch, JAX, and the Neuron compiler and runtime to enable and tune large-scale training workloads on the latest Trainium instances.

Requirements

3+ years of non-internship professional software development experience
2+ years of non-internship design or architecture experience (design patterns, reliability and scaling) of new and existing systems
Experience programming with at least one software programming language
3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
Bachelor's degree in computer science or equivalent

Technologies

PyTorch
JAX
AWS Neuron
Neuron compiler
Neuron runtime
Trainium
Trn3/Trn2/Trn1
Inf1/Inf2 servers

Benefits

Health insurance
401(k) matching
Paid time off
Parental leave
Sign-on payments
Restricted stock units (RSUs)
Basic Life & AD&D insurance
Supplemental life plans
Employee Assistance Program (EAP)
Mental Health Support
Medical Advice Line
Flexible Spending Accounts
Adoption and Surrogacy Reimbursement coverage

Description

Annapurna Labs designs silicon and software that accelerates innovation. Customers rely on us to create cloud solutions that solve challenges that were unimaginable a short time ago—even yesterday. Our custom chips, accelerators, and software stacks enable us to take on technical challenges that have never been seen before, and deliver results that help our customers change the world. AWS Neuron is the complete software stack for the AWS Trainium and Inferentia cloud-scale machine learning accelerators.

View full job page →

Distributed Training Developer Jobs

Machine Learning Engineer, Commerce Ads Ranking

Software Engineer- AI/ML, AWS Neuron Distributed Training - Performance Optimization

Get Distributed Training Developer Jobs alerts