Staff Software Engineer (Gen AI)

Job Description:

DataRobot delivers AI that maximizes impact and minimizes business risk. Our platform and applications integrate into core business processes so teams can develop, deliver, and govern AI at scale. DataRobot empowers practitioners to deliver predictive and generative AI, and enables leaders to secure their AI assets. Organizations worldwide rely on DataRobot for AI that makes sense for their business — today and in the future.

We are seeking an experienced Staff Software Engineer to join our GenAI team. The team is responsible for developing the core services and infrastructure that power DataRobot's Generative AI offerings, enabling our customers to build, deploy, and manage cutting-edge AI solutions. This role will involve designing and implementing scalable backend systems that support the full lifecycle of large language models (LLMs), from serving to monitoring and management. You will work on an internal platform to empower practitioners to deliver both predictive and generative AI.

Key Responsibilities:

• Lead the technical design and implementation of projects, driving them from conception to completion.

• Mentor other engineers, provide constructive feedback in code reviews, and promote a culture of team growth.

• Design and build robust, scalable application-level software and internal services that support GenAI products.

• Work on systems to ensure microservices are secure, performant, and can be deployed safely.

• Collaborate with product managers and data scientists to gather requirements and provide technical feedback on complexity and approaches.

• Engage in engineering on-call escalated support for services owned by the team.

• Troubleshoot complex production environments at the application and DB level.

Knowledge, Skills & Abilities:

• 8+ years of proven experience writing high-quality code in a collaborative environment using Python.

• Strong Computer Science fundamentals in object-oriented design, data structures, algorithm design, problem-solving, and complexity analysis.

• Deep experience with automated testing and test-driven development.

• Demonstrable knowledge of software architecture for large systems and real-world experience decoupling monolithic software into smaller reusable components.

• Expert proficiency in Kubernetes architecture and operations, with hands-on experience building and running software systems on Kubernetes clusters in production.

• Hands-on experience with infrastructure provisioning and configuration using Infrastructure as Code (IaC) principles.

• Proven track record of defining technical direction and influencing decisions without formal authority.

• Ability to solve ambiguous, open-ended challenges and influence technical direction across multiple teams.

• Great communication skills, with an ability to work in teams, share knowledge, and write documentation.

Nice to Have:

• Experience with AWS, Azure, and/or Google Cloud platforms.

• Experience with Docker and building containerized applications.

• CKAD (Certified Kubernetes Application Developer) certification.

• DevOps fundamentals and scripting experience.

• Experience supporting user-facing code and APIs.

• Experience with GPU clusters or multi-node AI/ML.

• Experience with Agentic AI.

The talent and dedication of our employees are at the core of DataRobot’s journey to be an iconic company. We strive to attract and retain the best talent by providing competitive pay and benefits with our employees’ well-being at the core. Here’s what your benefits package may include depending on your location and local legal requirements: Medical, Dental & Vision Insurance, Flexible Time Off Program, Paid Holidays, Paid Parental Leave, Global Employee Assistance Program (EAP) and more!

DataRobot Operating Principles:

• Wow Our Customers
• Set High Standards
• Be Better Than Yesterday
• Be Rigorous
• Assume Positive Intent
• Have the Tough Conversations
• Be Better Together
• Debate, Decide, Commit
• Deliver Results
• Overcommunicate

Research shows that many women only apply to jobs when they meet 100% of the qualifications while many men apply to jobs when they meet 60%. At DataRobot we encourage ALL candidates, especially women, people of color, LGBTQ+ identifying people, differently abled, and other people from marginalized groups to apply to our jobs, even if you do not check every box. We’d love to have a conversation with you and see if you might be a great fit.

DataRobot is proud to be an Equal Employment Opportunity and Affirmative Action employer. We do not discriminate based upon race, religion, color, national origin, gender (including pregnancy, childbirth, or related medical conditions), sexual orientation, gender identity, gender expression, age, status as a protected veteran, status as an individual with a disability, or other applicable legally protected characteristics. DataRobot is committed to working with and providing reasonable accommodations to applicants with physical and mental disabilities. Please see the United States Department of Labor’s EEO poster and EEO poster supplement for additional information.

All applicant data submitted is handled in accordance with our Applicant Privacy Policy.