Research Engineer - Generative AI - AI Frontiers

As a Research Engineer in AI Frontiers, you will design, develop, execute, and implement technology research projects in collaboration with other researchers, engineers, and product groups. As a member of a word-class research organization, you will be a part of research breakthroughs in the field and play a crucial role in developing, improving, and exploring the capabilities of Large Language Models (LLMs), reasoning and agentic AI. Embody our culture and values. Bachelors in Computer Science or relevant field AND 2+ years related experience OR Master's Degree in Computer Science or related field AND 1+ year(s) related experience OR Doctorate in Computer Science or related field OR equivalent experience 1+ year(s) experience developing with Python and Pytorch/JAX. Familiarity with architecture and optimizations for large language models. Hands-on work in debugging and profiling Pytorch distributed code. Basic understanding of working of CUDA kernels. Familiarity with pre-training, mid-training and/or post-training pipelines for language and/or multimodal models. Foundational understanding of reinforcement learning and key challenges in the field. Experience with verl, Ray, Megatron and/or vLLM is a significant plus. Any experience in building scalable services can be highly complementary.