ByteDance

Senior Software Engineer - AI Compute Infrastructure

ByteDance
Apply
16 hours ago

Responsibilities

  • Design and build large-scale container-based cluster management and orchestration systems with high performance, scalability, and resilience.
  • Architect cloud-native GPU and AI accelerator infrastructure for cost-efficient and secure ML platforms.
  • Collaborate across teams to deliver LLM inference solutions using vLLM, SGLang, TensorRT-LLM, and other inference engines.
  • Evaluate advances in Kubernetes, Ray, AI/ML, LLM infrastructure, open source, and systems research for production use.
  • Write maintainable, testable, scalable, production-ready code.

Requirements

  • Bachelor's or master's degree in Computer Science, Computer Engineering, or a related field, with 3+ years of relevant experience; Ph.D. candidates with strong systems or ML publications may also be considered.
  • Strong understanding of large-model inference, distributed and parallel systems, and/or high-performance networking systems.
  • Hands-on experience building cloud or ML infrastructure involving resource management, scheduling, request routing, monitoring, or orchestration.
  • Solid knowledge of Docker and Kubernetes.
  • Proficiency in at least one of Go, Rust, Python, or C++.
  • Preferred experience operating large-scale cluster management systems, including Kubernetes or Ray.
  • Preferred experience with workload scheduling, GPU orchestration, scaling, and isolation in production environments.
  • Preferred hands-on GPU programming experience with CUDA or experience with vLLM, SGLang, or TensorRT-LLM.
  • Preferred familiarity with AWS, Azure, GCP, SageMaker, Azure ML, or Vertex AI.
  • Preferred knowledge of Ray, DeepSpeed, PyTorch, and distributed training or inference platforms.

Benefits

  • Work with the open-source AIBrix project and contribute to the open-source community.
  • Collaborate with global, cross-functional engineering teams in a hyperscale, multi-cloud environment.
ByteDance

About ByteDance

10,000+ employees

ByteDance is a global incubator of platforms at the cutting edge of commerce, content, entertainment and enterprise services - over 2.5bn people interact with ByteDance products including TikTok. Creation is the core of ByteDance's purpose. Our products are built to help imaginations thrive. This is doubly true of the teams that make our innovations possible. Together, we inspire creativity and enrich life - a mission we aim towards achieving every day. At ByteDance, we create together and grow together. That's how we drive impact - for ourselves, our company, and the users we serve. We are committed to building a safe, healthy and positive online environment for all our users. We have over 110,000 employees based in more than 30 countries globally. Join us.

Contact me