Machine Learning Engineer - TRAE USDS
Is this job for you?
Build my CV Build your CV and see how well you match this role — and every other one.
The role
Design, train, and fine-tune large language models (LLMs) to support the product’s core reasoning and code generation capabilities.
Build efficient, stable, and scalable model training and evaluation pipelines.
Collaborate with infrastructure and product teams to deploy and monitor models efficiently on GPU clusters.
Continuously optimize models for latency, throughput, and accuracy, applying cutting-edge techniques in model optimization and inference acceleration.
Stay up to date with the latest advancements in large-model training and deployment, and translate them into production-ready solutions.
This role operates within a hybrid work model and requires effective collaboration across cross-functional teams.
Build efficient, stable, and scalable model training and evaluation pipelines.
Collaborate with infrastructure and product teams to deploy and monitor models efficiently on GPU clusters.
Continuously optimize models for latency, throughput, and accuracy, applying cutting-edge techniques in model optimization and inference acceleration.
Stay up to date with the latest advancements in large-model training and deployment, and translate them into production-ready solutions.
This role operates within a hybrid work model and requires effective collaboration across cross-functional teams.
See the full job post
Responsibilities, requirements, skills and benefits — create your free account.
or
Already have an account?
Log inSimilar openings
Other roles that could suit you.
? Intern, Machine Learning Engineer ? Senior Data Scientist - Product Analytics (Payments/Ecommerce) ? Data Scientist Graduate (TikTok-Product-Data Science)-2026 Start (PhD) ? [HackerRank] Software Engineer Graduate (On-Device AI) - 2025 Start (BS/MS) ? Senior Director, Risk Data Science ? Data Scientist Graduate (TikTok-Product-Data Science)-2026 Start (BS/MS)
Remote workPartial
CitySan Jose, United States