Staff Software Engineer - Agent Quality

STAFF SOFTWARE ENGINEER - AI RESEARCH INFRASTRUCTURE P-1215 At Databricks, we are obsessed with enabling data teams to solve the world’s toughest problems, from security threat detection to cancer drug development. We do this by building and running the world’s best data and AI platform so our customers can focus on the high-value challenges that are central to their own missions. The Databricks AI Research organization enables companies to develop AI models and agents using their own data, with technologies ranging from post-training open source LLMs to developing advanced multi-agent architectures. Databricks AI does so by producing novel science and putting it into production. Databricks AI is committed to the belief that a company’s AI models and agents are just as valuable as any other core IP, and that high-quality AI should be available to all. Job Description As a Staff Software Engineer, AI Research Infrastructure, you will be developing and running the research stack that powers Databricks AI Research. You will design and build services that schedule, orchestrate, and observe large‑scale training and inference experiment workloads across thousands of GPUs, improve our dev tooling and ensure that researchers can iterate quickly without sacrificing reliability, efficiency, or security. You’ll partner closely with research scientists, ML engineers, and platform teams to turn experimental workloads into robust, repeatable pipelines, and to push the limits of what our infrastructure can support. The Impact you will have As a Staff Software Engineer on the AI Research Infra Team at Databricks, you will:  * Design and implement infrastructure that supports large‑scale experiments, data processing, and model training (e.g., HPC clusters, GPU fleets, or cloud‑based systems) * Enable researchers to go from idea to large‑scale experiment in minutes, not days, by building powerful abstractions for job submission, scheduling, and monitoring. * Create tooling that improves research developer productivity, such as experiment management systems, CI/testing infrastructure for research code, and workflows that reduce iteration time. * Influence the long‑term roadmap for research computation, shaping how Databricks AI Research train, evaluate, and ship models to customers. * Serve as a technical mentor and force multiplier for other engineers working on compute, infra, and AI systems. What We Look for * BS/MS or PhD in Computer Science or related field * 5+ years of software engineering experience, including substantial time working on large‑scale distributed systems or infrastructure. * Have deep experience with building and operating distributed systems, data pipelines, or large‑scale backend services, ideally involving GPUs, clusters, or major cloud providers. * Are proficient in one or more systems programming languages (e.g., C++, Rust, Go, Java, Scala) and can design, implement, and debug complex services. * Have built or significantly contributed to cluster schedulers, resource managers, or large‑scale job orchestration systems (e.g., Kubernetes, Slurm, Ray, custom internal systems). * Understand modern ML training and inference workflows (e.g., distributed training, model parallelism, fine‑tuning, evaluation), even if you’re not primarily a research scientist. * Can move fast and be pragmatic in getting things done, while caring about operational excellence. Have driven complex systems from prototype to stable, well‑owned services. * Communicate clearly with both researchers and engineers, and enjoy translating between research needs and infra realities.   Pay Range Transparency Databricks is committed to fair and equitable compensation practices. The pay range(s) for this role is listed below and represents the expected salary range for non-commissionable roles or on-target earnings for commissionable roles.  Actual compensation packages are based on several factors that are unique to each candidate, including but not limited to job-related skills, depth of experience, relevant certifications and training, and specific work location. Based on the factors above, Databricks anticipates utilizing the full width of the range. The total compensation package for this position may also include eligibility for annual performance bonus, equity, and the benefits listed above. For more information regarding which range your location is in visit our page here.   Local Pay Range $190,000—$270,000 USD About Databricks Databricks is the data and AI company. More than 10,000 organizations worldwide — including Comcast, Condé Nast, Grammarly, and over 50% of the Fortune 500 — rely on the Databricks Data Intelligence Platform to unify and democratize data, analytics and AI. Databricks is headquartered in San Francisco, with offices around the globe and was founded by the original creators of Lakehouse, Apache Spark™, Delta Lake and MLflow. To learn more, follow Databricks on Twitter, LinkedIn and Facebook. Benefits At Databricks, we strive to provide comprehensive benefits and perks that meet the needs of all of our employees. For specific details on the benefits offered in your region click here. Our Commitment to Diversity and Inclusion At Databricks, we are committed to fostering a diverse and inclusive culture where everyone can excel. We take great care to ensure that our hiring practices are inclusive and meet equal employment opportunity standards. Individuals looking for employment at Databricks are considered without regard to age, color, disability, ethnicity, family or marital status, gender identity or expression, language, national origin, physical and mental ability, political affiliation, race, religion, sexual orientation, socio-economic status, veteran status, and other protected characteristics. Compliance If access to export-controlled technology or source code is required for performance of job duties, it is within Employer's discretion whether to apply for a U.S. government license for such positions, and Employer may decline to proceed with an applicant on this basis alone.

We’re looking for a Staff Software Engineer to integrate and improve our AI developer experience so that engineers at Asana can use AI to increase their velocity. As part of the AI Developer Productivity team, you’ll set technical direction and build the next generation of AI-powered developer tools across editors, IDEs, CLIs, code review, and cloud and local coding agents. This role is based in our New York City office with an office-centric hybrid schedule. The standard in-office days are Monday, Tuesday, and Thursday. Most Asanas have the option to work from home on Wednesdays. Working from home on Fridays depends on the type of work you do and the teams with which you partner. If you're interviewing for this role, your recruiter will share more about the in-office requirements. WHAT YOU’LL ACHIEVE * Design and build AI-augmented workflows and systems that help coding agents understand the codebase, follow engineering practices, and enable Asana engineers to complete software development tasks faster and with more confidence. * Design and scale autonomous cloud agents that take on complex, multi-step engineering tasks to reduce toil and enable engineers to focus on higher-leverage work. * Build and refine IDE, editor, and CLI integrations that make AI-assisted development feel intuitive in engineers' daily work. * Create reusable agent skills, tools, context, and integrations that teams across Asana can build on rather than reinvent. * Improve AI-assisted code review workflows so engineers get faster, higher-quality feedback before and during review. * Drive adoption of AI developer tools across engineering through usability improvements, measurement, documentation, and enablement. * Set technical direction for the team, balancing experimentation with reliability, maintainability, and long-term platform thinking. * Partner cross-functionally with teams across the engineering organization to understand engineering needs, identify workflow friction, and scale high-impact solutions. ABOUT YOU * 8+ years of software engineering experience, including 3+ years of experience leading complex technical projects across teams. * Experience building developer tools, internal platforms, infrastructure, IDE/editor integrations, CLIs, or workflow automation. * Hands-on experience with AI-assisted development tools (such as Cursor, Claude Code, or Codex), or extending coding agents, MCP-based integrations, AI-powered code review systems, or agent skill/tool ecosystems. * Experience building evaluation and quality frameworks for AI-generated outputs. * Strong technical judgment and systems design skills, with the ability to make pragmatic tradeoffs in ambiguous or rapidly changing environments. * Demonstrates curiosity about AI tools and emerging technologies, with a willingness to learn and leverage them to enhance productivity, collaboration, or decision-making. WHAT WE’LL OFFER Our comprehensive compensation package plays a big part in how we recognize you for the impact you have on our path to achieving our mission. We believe that compensation should be reflective of the value you create relative to the market value of your role. To ensure pay is fair and not impacted by biases, we're committed to looking at market value which is why we check ourselves and conduct a yearly pay equity audit. For this role, the estimated base salary range is between $248,000 - $282,000. The actual base salary will vary based on various factors, including market and individual qualifications objectively assessed during the interview process. The listed range above is a guideline, and the base salary range for this role may be modified. In addition to base salary, your compensation package may include additional components such as equity, sales incentive pay (for most sales roles), and benefits. If you're interviewing for this role, speak with your Talent Acquisition Partner to learn more about the total compensation and benefits for this role. We strive to provide equitable and competitive benefits packages that support our employees worldwide and include: * Mental health, wellness & fitness benefits * Career coaching & support * Inclusive family building benefits * Long-term savings or retirement plans * In-office culinary options to cater to your dietary preferences These are just some of the benefits we offer, and benefits may vary based on role, country, and local regulations. If you're interviewing for this role, speak with your Talent Acquisition Partner to learn more about the total compensation and benefits for this role. #LI-Hybrid #LI-AS2 About us Asana is a leading platform for human + AI collaboration. Millions of teams around the world rely on Asana to achieve their most important goals, faster. Asana has been named to Fortune's Best Workplaces for 7+ years and recognized by Fast Company, Forbes, and Gartner for excellence in workplace culture and innovation. We offer an exceptional office-centric culture while adopting the best elements of hybrid models to ensure that every one of our global team members can work together effortlessly. With 13+ offices all over the world, we are always looking for individuals who care about building technology that drives positive change in the world and a culture where everyone feels that they belong. Join Asana’s Talent Network to stay up to date on job opportunities and life at Asana.