Senior Software Engineer - Feature Store (AI Platform)

At Datadog, we're building a unified AI Platform that powers ML and LLM applications across our product suite. The Feature Store team is central to this mission-delivering infrastructure that allows applied scientists and engineers to build, version, and serve features at scale across both offline training and real-time inference.

Our feature store supports complex data pipelines, embedding storage, and feature discoverability in multi-region production environments. It integrates deeply with our training, experimentation, and model serving workflows, ensuring high availability, low-latency access, and traceable lineage for every feature used in the ML lifecycle.

We're looking for a Senior Software Engineer to design and build the next generation of Datadog's feature infrastructure. You will work at the intersection of data engineering, ML infrastructure, and distributed systems to deliver scalable, low-friction solutions used across teams.

At Datadog, we place value in our office culture - the relationships that it builds, the creativity it brings to the table, and the collaboration of being together. We operate as a hybrid workplace to ensure our employees can create a work-life harmony that best fits them.

What You'll Do:

• Design and build distributed feature pipelines that power both training and low-latency inference use cases.
• Implement feature versioning, cataloging, and access management capabilities, ensuring traceability and governance across teams.
• Drive the integration of embedding storage systems (e.g., DDVector) to support scalable training and real-time retrieval.
• Build tooling to surface feature metadata, dependencies, and usage analytics to empower discovery and reuse.
• Partner with training, serving, and registry teams to deliver seamless integration across the AI Platform lifecycle.
• Own the operational excellence of your systems through monitoring, performance optimization, and cost efficiency.

Who You Are:

• You have 6+ years of experience as a data or platform engineer, including designing large-scale distributed systems for ML or analytics workloads.
• You've built and operated data pipelines or storage systems with technologies like Kafka, Snowflake, ClickHouse, dbt, or similar.
• You are proficient in Python, Scala, Go, or Java, and have production experience working in cloud environments (e.g., AWS).
• You understand data model design and storage formats optimized for ML workflows (e.g., columnar formats, LSM trees).
• You value cross-functional collaboration and can work closely with applied scientists, backend engineers, and infra teams.
• Bonus: experience building or operating a feature store, embedding database, or metadata management platform.

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.

Benefits and Growth:

• New hire stock equity (RSUs) and employee stock purchase plan (ESPP)
• Continuous professional development, product training, and career pathing
• Intradepartmental mentor and buddy program for in-house networking
• An inclusive company culture, ability to join our Community Guilds (Datadog employee resource groups)
• Access to Inclusion Talks, our internal panel discussions
• Free, global mental health benefits for employees and dependents age 6+
• Competitive global benefits

Benefits and Growth listed above may vary based on the country of your employment and the nature of your employment with Datadog.

About Datadog:

Datadog (NASDAQ: DDOG) is a global SaaS business, delivering a rare combination of growth and profitability. We are on a mission to break down silos and solve complexity in the cloud age by enabling digital transformation, cloud migration, and infrastructure monitoring of our customers' entire technology stacks. Built by engineers, for engineers, Datadog is used by organizations of all sizes across a wide range of industries. Together, we champion professional development, diversity of thought, innovation, and work excellence to empower continuous growth. Join the pack and become part of a collaborative, pragmatic, and thoughtful people-first community where we solve tough problems, take smart risks, and celebrate one another. Learn more about #DatadogLife on Instagram, LinkedIn, and Datadog Learning Center.

Equal Opportunity at Datadog:

Datadog is an Affirmative Action and Equal Opportunity Employer and 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 more. We also consider qualified applicants regardless of criminal histories, consistent with legal requirements. Here are our Candidate Legal Notices for your reference.

Your Privacy:

Any information you submit to Datadog as part of your application will be processed in accordance with Datadog's Applicant and Candidate Privacy Notice.