Machine Learning Engineer, AI Coding Tools

ByteDance
San Jose
Workplace: OnsiteFull timeFunction: Data Science & Machine LearningEducation: bachelorsSkills: []

Join TRAE’s core R&D team to train, fine-tune, and deploy large language models that power end-to-end AI software generation. Build scalable training and evaluation pipelines, collaborate with infrastructure and product teams to deploy on GPU clusters, and continuously optimize models for latency, throughput, and accuracy. Apply state-of-the-art model optimization and inference acceleration techniques to advance TRAE’s intelligent engineering capabilities.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
ByteDance
ByteDance
1 month ago

Machine Learning Engineer, AI Coding Tools

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 31 days agoStatus: Live
Reposted: similar role first listed 1 month ago

Job Summary

Join TRAE’s core R&D team to train, fine-tune, and deploy large language models that power end-to-end AI software generation. Build scalable training and evaluation pipelines, collaborate with infrastructure and product teams to deploy on GPU clusters, and continuously optimize models for latency, throughput, and accuracy. Apply state-of-the-art model optimization and inference acceleration techniques to advance TRAE’s intelligent engineering capabilities.
Location: San Jose
Workplace: Onsite
Employment Type: Full time
Job Function: Data Science & Machine Learning

Key Responsibilities

  • •Design, train, and fine-tune large language models (LLMs) for reasoning and code generation.
  • •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 and apply cutting-edge techniques for large model optimization and inference acceleration.

Key Requirements

  • •Bachelor’s degree or above in Computer Science, Electrical Engineering, Mathematics, or a related field.
  • •Solid foundation in machine learning, deep learning, and optimization algorithms.
  • •Proficiency in PyTorch or TensorFlow, with programming skills in Python and C++/CUDA.
  • •Experience with large-scale distributed training, mixed-precision training, or model parallelism.
  • •Hands-on experience with model quantization, pruning, or CUDA-based deployment optimization.
Experience:Machine learningDeep learningLLMDistributed trainingAI systems
Education:Bachelor's
Tech Stack:PyTorchTensorFlowPythonC++CUDAGPU clustersLLMsQuantizationPruningInference acceleration

Company Brief

ByteDance
Develops consumer internet and content platforms, including TikTok and other apps for short-form video, news, and entertainment. It also builds advertising, commerce, and creator tools that connect audiences, brands, and publishers across global markets.
Industry: Digital Media
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Established Company
Headquarters: Beijing, China
Founded: 2012
WebsiteLinkedIn