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 that enable robust reasoning and code generation within an end-to-end AI software development workflow.
Build efficient, scalable training and evaluation pipelines for LLMs on GPU clusters.
Collaborate with infrastructure and product teams to deploy, monitor, and maintain production-ready models.
Continuously optimize for latency, throughput, and accuracy using state-of-the-art techniques in model optimization and inference acceleration.
Stay up-to-date with the latest developments in distributed training, quantization, pruning, and deployment strategies.
Required proficiency in PyTorch or TensorFlow, Python and C++/CUDA, with hands-on experience in large-scale training and model deployment.
Build efficient, scalable training and evaluation pipelines for LLMs on GPU clusters.
Collaborate with infrastructure and product teams to deploy, monitor, and maintain production-ready models.
Continuously optimize for latency, throughput, and accuracy using state-of-the-art techniques in model optimization and inference acceleration.
Stay up-to-date with the latest developments in distributed training, quantization, pruning, and deployment strategies.
Required proficiency in PyTorch or TensorFlow, Python and C++/CUDA, with hands-on experience in large-scale training and model deployment.
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)