Senior AI Research Engineer - GPU Systems & Kernels

About the job

As a Research Engineer on the Frontier Foundations team, you will partner with Research Scientists to make large-scale model training and inference as fast and efficient as the hardware allows. You will work at the lowest level of our stack, writing and tuning GPU kernels and removing the memory and communication bottlenecks that limit large Mixture-of-Experts models on the latest GPUs. Building on our track record of AI-powered solutions (e.g., Bits AI, Bits Evolve, and our time series foundation model Toto), Datadog AI Research (DAIR) tackles high-risk, high-reward problems grounded in real-world challenges in cloud observability and security. Frontier Foundations builds and maintains the…