Research 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 LLM 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 training frameworks such as Pytorch.

Nice to have

• GPU high-performance optimization experience, including CUDA, computer architecture, parallel computing, memory access, or low-bit computing.
• Familiarity with distributed training and inference technologies; LLM knowledge or model-optimization experience.

Tools

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

AI focus

Build and optimize systems for LLM and AI-agent training, inference, and reinforcement learning, including scalable workflows for dynamic agent workloads.