LLM Engineer for AI Software Development Tools
Summary
Build and deploy large language models for code generation and reasoning using PyTorch/TensorFlow, distributed training, and GPU clusters.
ByteDance's TRAE team is hiring a Machine Learning Engineer to design, train, and deploy LLMs that support code generation and reasoning. You will build robust training and evaluation pipelines and collaborate with infrastructure and product teams to run models on GPU clusters.
You should have a strong ML background, experience with PyTorch or TensorFlow, Python and C++/CUDA, and hands-on work with distributed training, quantization, or pruning.