ByteDance
Machine Learning Engineer, AI Coding Tools
San Jose, California · Full-time · Posted Oct 28, 2026
- Education
- Bachelor’s degree or above in Computer Science, Electrical Engineering, Mathematics, or a related field.
About the job
Train, optimize, and deploy large language models that power TRAE’s reasoning and code-generation capabilities.
What you'll do
- Design, train, and fine-tune LLMs and build scalable training and evaluation pipelines.
- Deploy and monitor models on GPU clusters with infrastructure and product teams.
- Optimize models for latency, throughput, and accuracy, applying model-optimization and inference-acceleration techniques.
What they ask for
- Bachelor’s degree or above in Computer Science, Electrical Engineering, Mathematics, or a related field.
- Foundation in machine learning, deep learning, and optimization algorithms; proficiency in PyTorch or TensorFlow, 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.
Nice to have
- PhD in AI-related research fields.
- Hands-on experience developing and deploying AI systems in a real-world setting.
Tools
PyTorch, TensorFlow, Python, C++, CUDA
AI focus
LLM training, fine-tuning, optimization, quantization, inference acceleration, and deployment for software engineering, reasoning, and code generation.
Model-written summary · the original posting is authoritative