Software Engineer

About the Role

H2O.ai is seeking a full-stack Software Engineer to join their Predictive AI engineering team. This role supports work with U.S. federal government customers and involves developing H2O Driverless AI, H2O-3, and Sparkling Water, the cloud platform AutoML and Predictive stack. The position is remote and based in the United States.

Key Responsibilities

• Design, implement, and maintain backend services and APIs in Python, Go, and Java.
• Develop and operate cloud-native services on Kubernetes for deploying and running predictive AI workloads in cloud and on-premise environments.
• Build and maintain CI/CD pipelines, multi-architecture builds (x86 and ARM), dependency upgrades, and vulnerability remediation, including FedRAMP-compliant builds.
• Build and maintain web interfaces using React and TypeScript.
• Work with support and field engineering to resolve customer issues, including for federal customers, and drive fixes back into the product.
• Translate product requirements into technical specifications and make architectural trade-offs transparent.
• Drive engineering excellence through rigorous code reviews, comprehensive testing, and clear documentation.
• Monitor application performance, troubleshoot issues, and optimize systems for reliability and efficiency.
• Collaborate with data science, product, and customer-facing teams.
• Contribute features and fixes across Driverless AI, H2O-3, Sparkling Water, and their client libraries.

Requirements & Tech Stack

• Bachelor’s or Master’s degree in Computer Science or Engineering, or equivalent practical experience.
• Professional, internship, or project experience building and shipping software.
• Proficiency in at least one backend language: Python, Go, or Java.
• Familiarity building web applications with React and TypeScript, or a similar modern frontend framework.
• Strong computer science fundamentals, including data structures, concurrency, networking, performance, and debugging.
• Hands-on experience with CI/CD pipelines (GitHub Actions, Jenkins).
• Working knowledge of containers and cloud-native deployment (Docker, Kubernetes, AWS, GCP, or Azure).
• Experience designing and building REST or gRPC APIs and working with relational databases such as PostgreSQL.
• Clear written and verbal communication with technical peers, customers, and non-engineering stakeholders.
• Ability to thrive in ambiguous environments and proactively take ownership of technical challenges.

Our Stack:

• Backend: Python, Go, Java
• Frontend: React and TypeScript
• Platforms: Driverless AI, H2O-3, Sparkling Water
• Infrastructure: Kubernetes, Docker, Helm, multi-cloud (AWS, GCP, Azure)
• Data: PostgreSQL, object storage, message queues
• CI/CD: Jenkins, GitHub Actions

Prior knowledge of every tool is not required. Ability to learn and adapt quickly is valued.

Qualifications (How to Stand Out)

• ML/AI experience, including training, serving, or scoring models in production.
• Experience in regulated or air-gapped environments.
• An active U.S. government Public Trust or security clearance.
• Experience with Driverless AI, H2O-3, or Sparkling Water, or contributions to open-source machine learning frameworks.
• Experience with Apache Spark or other distributed computing systems.
• Familiarity with LLM-based or agentic systems.
• Experience building an SDK or library used by other developers.