Senior AI Research Engineer - GPU Systems & Kernels

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 high-performance infrastructure that lets us train and serve large language models at scale. It supports both of our research areas:

Trained Agents for Observability - Post-training models to operate autonomously across Datadog's domain. SRE incident response is our first target, with a clear path to code repair, security response, and infrastructure optimization. We build the simulation environments, RL training loops, and evaluation infrastructure needed to train agents that match or surpass frontier models at a fraction of the cost.

What You'll Do:

• Write, tune, and debug GPU kernels (CUDA, Triton, or similar) for the operations that dominate LLM training and inference.
• Optimize for the latest NVIDIA GPU generations, including memory hierarchy, tensor cores, and low-precision formats such as FP8 and FP4
• Reduce communication cost in distributed workloads (all-to-all, all-reduce) and overlap communication with compute across NVLink and across nodes
• Profile end-to-end training and rollout workloads to identify whether they are compute-, memory-, or communication-bound, and quantify the headroom
• Build benchmarks, profiling tools, and regression tests that guide optimization decisions
• Collaborate with Research Scientists to turn new model architectures into efficient implementations
• Explore emerging techniques such as low-precision training and serving, speculative decoding, and fused or persistent kernels

Who You Are:

• You have depth in at least one of: GPU kernel development and optimization (CUDA or similar), low-latency communication and collective operations (e.g., RDMA, NCCL), or large-scale training infrastructure
• You can reason about hardware constraints (memory, bandwidth, interconnect topology) and explain where a bottleneck is and how you would confirm it
• You are proficient in a systems language (e.g., C++, Rust, or CUDA C) and in Python
• You have strong debugging and profiling skills, including tools such as Nsight
• You can explain design and performance trade-offs clearly to both technical and non-technical audiences

Bonus Points (any of the following):

• You have experience with Mixture-of-Experts models, expert parallelism, or sparse model inference
• You are familiar with GPU communication libraries such as NCCL or NVSHMEM, and with compute/communication overlap
• You have experience with Triton, CUTLASS, or similar kernel frameworks
• You are familiar with distributed training frameworks such as Megatron-LM, TorchTitan, or DeepSpeed
• You have built GPU-intensive systems in graphics, rendering, simulation, or scientific computing
• You have contributed to open-source ML or systems infrastructure projects

Datadog values people from all walks of life. We understand not everyone will meet all the above qualifications on day one. That's okay. If you're passionate about technology and want to grow your skills, we encourage you to apply.

#LI-Hybrid

About Datadog:

Datadog is the leading observability and security platform for the AI era, providing businesses with unified visibility across the technology stack to manage complexity at scale. It brings applications, infrastructure, data, models, and security into one place, using AI to detect and resolve issues before they impact customers. Trusted globally by Fortune 500 companies and high-growth AI leaders, Datadog enables businesses to move faster with clarity and confidence. Learn more about #DatadogLife on Instagram, LinkedIn, and Datadog Learning Center.

Equal Opportunity at Datadog:

Datadog is proud to offer equal employment opportunity to everyone regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity, veteran status, and other characteristics protected by law. We also consider qualified applicants regardless of criminal histories, consistent with legal requirements. Here are our Candidate Legal Notices for your reference.

Datadog endeavors to make our Careers Page accessible to all users. If you would like to contact us regarding the accessibility of our website or need assistance completing the application process, please complete this form. This form is for accommodation requests only and cannot be used to inquire about the status of applications.

Privacy and AI Guidelines:

Any information you submit to Datadog as part of your application will be processed in accordance with Datadog’s Applicant and Candidate Privacy Notice. For information on our AI policy, please visit Interviewing at Datadog AI Guidelines.