Applied Scientist - LLM Training System as a Service - Global Frontier Tech Recruitment Program - 2027 Start (PhD)

What you'll do

• Develop and optimize LLM training, inference, and reinforcement-learning frameworks.
• Work with model researchers to scale LLM training and reinforcement learning.
• Optimize GPU and CUDA performance for training, inference, and reinforcement-learning engines.

What they ask for

• Currently pursuing a PhD in computer science, automation, electronics engineering, or a related technical discipline.
• Proficient in algorithms and data structures and familiar with Python.
• Understand deep-learning principles, neural-network architectures, and frameworks such as Pytorch.

Nice to have

• GPU high-performance computing optimization with CUDA; knowledge of computer architecture and parallel, memory-access, or low-bit optimization.
• Familiarity with FSDP, Deepspeed, JAX SPMD, Megatron-LM, Verl, TensorRT-LLM, ORCA, VLLM, or SGLang; LLM knowledge or model-acceleration experience.

Tools

Python, Pytorch, CUDA, FSDP, Deepspeed, JAX SPMD, Megatron-LM, TensorRT-LLM

AI focus

Build and optimize LLM and AI Agent training and inference systems, including reinforcement learning and GPU acceleration.