Research Scientist Graduate (Applied Machine Learning - ML System) - 2026 Start (PhD)

What you'll do

• Develop and optimize LLM training, inference, and RL frameworks.
• Collaborate with model researchers to scale LLM training and RL.
• Optimize GPU and CUDA performance for LLM training, inference, and RL engines.

What they ask for

• Completing or recently completed a PhD in Software Development, Computer Science, Computer Engineering, or a related technical discipline.
• Proficiency in algorithms and data structures; familiarity with Python.
• Understand deep learning fundamentals, neural network architecture, and frameworks such as Pytorch.

Nice to have

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

Tools

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

AI focus

Build and optimize machine-learning infrastructure for LLM training, inference, and RL, including performance optimization and scaling.