Senior Software Engineer

Founded in 2012, H2O.ai is dedicated to democratizing AI. As the world’s leading agentic AI company, H2O.ai converges Generative and Predictive AI to help enterprises and public sector agencies develop purpose-built GenAI applications on their private data. Its open-source technology is trusted by over 20,000 organizations worldwide, including more than half of the Fortune 500. H2O.ai powers AI transformation for companies such as AT&T, Commonwealth Bank of Australia, Singtel, Chipotle, Workday, Progressive Insurance, and the NIH.

H2O.ai partners include Dell Technologies, Deloitte, Ernst & Young (EY), NVIDIA, Snowflake, AWS, Google Cloud Platform (GCP), and VAST. H2O.ai’s AI for Good program supports nonprofit groups, foundations, and communities in advancing education, healthcare, and environmental conservation. With a vibrant community of 2 million data scientists worldwide, H2O.ai aims to co-create valuable AI applications for all users.

About This Opportunity

Join the team building H2O Driverless AI, our award-winning Automated Machine Learning (AutoML) platform. Driverless AI is a flagship product designed to use AI to make AI, automating the complex data science pipeline—including intelligent feature engineering, model selection, model validation, and interpretability (MLI). The platform significantly accelerates the time from data to production-ready models, generating highly optimized scoring pipelines in both Python and Java for high-performance, low-latency deployment.

As a Senior Software Engineer, you will play a critical role in the full lifecycle development of core platform features, from high-performance computation back-end services to user-facing components.

This position is based in Sri Lanka.

What You Will Do

• Design, develop, and maintain core components of the H2O Driverless AI platform, focusing on scalability, performance, and reliability.
• Write clean, efficient, and well-documented code primarily in Python, contributing to the core AutoML engine, feature transformers, and model pipelines.
• Build and improve user interface elements and complex data visualizations using TypeScript.
• Implement and manage deployment mechanisms using Docker and containerization best practices to ensure seamless deployment across cloud and on-premise environments.
• Collaborate closely with data scientists, researchers, and product managers to translate complex machine learning research into production-quality software features.
• Participate in code reviews, design discussions, and mentor junior engineers on software architecture and engineering best practices.

What We Are Looking For

• 4+ years of professional experience in software development, building and shipping scalable enterprise software products.
• Deep expertise and fluency in Python (required for primary product development).
• Solid experience developing enterprise applications using Java and/or working with Java-based APIs and systems.
• Proven experience with front-end development using TypeScript or JavaScript.
• Expertise in containerization and orchestration, specifically utilizing Docker in development and production environments.
• Strong understanding of distributed systems, multi-threading, concurrency, and performance optimization techniques.

How To Stand Out from the Crowd

• Hands-on experience or strong theoretical background in Machine Learning (ML), Deep Learning, or Data Science concepts (e.g., feature engineering, model training/testing, explainability techniques).
• Familiarity with ML frameworks (e.g., TensorFlow, PyTorch, XGBoost) or distributed computing technologies (e.g., Dask, Spark).

Why H2o.Ai?

• Market Leader in Total Rewards
• Remote-Friendly Culture
• Flexible working environment
• Be part of a world-class team
• Career Growth

H2O.ai is committed to creating a diverse and inclusive culture. All qualified applicants will receive consideration for employment without regard to their race, ethnicity, religion, gender, sexual orientation, age, disability status, or any other protected characteristic under applicable law.

H2O.ai is an innovative AI cloud platform company, leading the mission to democratize AI for everyone. Thousands of organizations worldwide have utilized our cutting-edge technology across various industries. We’ve made it easy for people at all levels to generate breakthrough solutions to complex business problems and advance the discovery of new ideas and revenue streams. We push the boundaries of what is possible with artificial intelligence. 

H2O.ai employs the world’s top Kaggle Grandmasters, the community of best-in-the-world machine learning practitioners and data scientists. A strong AI for Good ethos and a responsible approach to AI drive the company’s purpose.

Please visit www.H2O.ai to learn more.

#LI-Hybrid