Senior Software Engineer, AI Agent jobs in United States
cer-icon
Apply on Employer Site
company-logo

Astronomer · 1 month ago

Senior Software Engineer, AI Agent

Astronomer empowers data teams to bring mission-critical software, analytics, and AI to life and is the company behind Astro, the industry-leading unified DataOps platform powered by Apache Airflow®. In this role, you will design and build intelligent systems for data engineering, collaborate with R&D and product teams, and influence the technical vision for AI-driven data products.

AnalyticsBig DataData IntegrationMachine LearningSoftware
check
Comp. & Benefits
check
H1B Sponsor Likelynote

Responsibilities

Shape the future of AI for data engineering - build intelligent systems that understand, reason about, and optimize the flow of data across entire organizations
Design and engineer the brain of Astronomer’s context layer, crafting components that power data modeling, semantic search, retrieval, and code generation
Push the boundaries of applied AI - experiment with LLMs, embeddings, and cutting-edge retrieval techniques to create developer tools that deliver insights to you and our customers
Turn research into reality - work side by side with R&D and product teams to bring early AI concepts to life in the product experience
Solve high-impact information retrieval and search challenges at a global scale, leveraging Astronomer’s unparalleled visibility into data pipelines across industries
Influence the technical vision and architecture for the next generation of AI-driven data products
Represent Astronomer in the community - through open-source contributions, technical talks, and publications that showcase our leadership in AI and data innovation

Qualification

LLMsApache AirflowVector databasesEmbeddingsInformation retrievalSearchCollaboration skillsCommunication skillsCreative mindset

Required

Empathy for users, and a deep interest in improving the workflows of data professionals
Familiarity with early-stage product development; comfortable working with ambiguity in a fast-changing field
Experience with LLMs, vector databases, embeddings, or other applied AI areas - or a strong desire to dive in
Strong collaboration and communication skills - you can explain complex systems clearly to both technical and non-technical audiences
A creative, experimental mindset: you enjoy exploring uncharted areas, validating hypotheses, and learning through iteration
A collaborative approach and comfort working in an evolving, research-driven environment where ideas move quickly

Preferred

A passion for AI systems for data, developer tools, or machine learning infrastructure
Familiarity with Apache Airflow or other orchestration tools
Demonstrated contributions to open source projects
Experience in search, IR, or large-scale data infrastructure
Exposure to early-stage startups or R&D organizations where ambiguity is the norm

Benefits

Equity component
Comprehensive benefits package

Company

Astronomer

twittertwittertwitter
company-logo
Astronomer is a data orchestration platform built on Apache Airflow, enabling teams to build, run, and monitor data pipelines at scale.

H1B Sponsorship

Astronomer has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (3)
2024 (3)
2023 (5)
2022 (9)
2021 (1)
2020 (4)

Funding

Current Stage
Late Stage
Total Funding
$377.8M
Key Investors
Bain Capital VenturesInsight PartnersSierra Ventures
2025-05-01Series D· $93M
2022-03-23Series C· $213M
2021-01-29Series B· $43.6M

Leadership Team

leader-logo
Pete DeJoy
CEO, Co-Founder
linkedin
leader-logo
Steven Hillion
VP, Data & ML
linkedin
Company data provided by crunchbase