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 to support advanced code generation and reasoning capabilities.
Build efficient, stable, and scalable model training and evaluation pipelines.
Collaborate with infrastructure and product teams to deploy and monitor models on GPU clusters.
Continuously optimize models for latency, throughput, and accuracy.
Stay up-to-date with cutting-edge techniques in large-model optimization and inference acceleration.
Qualifications include a degree in CS/EE/Math, strong ML fundamentals, PyTorch or TensorFlow, Python and C++/CUDA, distributed training experience, and quantization or pruning skills; Master's and 3+ years are preferred.
Build efficient, stable, and scalable model training and evaluation pipelines.
Collaborate with infrastructure and product teams to deploy and monitor models on GPU clusters.
Continuously optimize models for latency, throughput, and accuracy.
Stay up-to-date with cutting-edge techniques in large-model optimization and inference acceleration.
Qualifications include a degree in CS/EE/Math, strong ML fundamentals, PyTorch or TensorFlow, Python and C++/CUDA, distributed training experience, and quantization or pruning skills; Master's and 3+ years are preferred.
See the full job post
Responsibilities, requirements, skills and benefits — create your free account.
Already have an account? Log in
Similar 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)