Senior Machine Learning Engineer

Founded in 2012, H2O.ai is at the forefront of the AI movement to democratize Generative AI. H2O.ai’s open-source Generative AI and Enterprise h2oGPT, combined with Document AI and the award-winning autoML Driverless AI, have transformed more than 20,000 global organizations and over half of the Fortune 500 and household brands, including AT&T, Commonwealth Bank of Australia, Chipotle, ADP, Workday, Progressive Insurance, and AES.

Our “AI for Good” program supports nonprofit groups, foundations, and communities in their efforts to advance education, healthcare, and environmental conservation, including identifying areas vulnerable to natural disasters and protecting endangered species. 

We have a vibrant community of two million data scientists worldwide and aim to bring together the world’s top data scientists with customers to co-create GenAI applications that are usable and valuable by everyone. Business users can now leverage the power of LLMs to enhance productivity with enterprise applications.

H2O.ai is a Visionary in the 2024 Gartner® Magic Quadrant™ for Data Science and Machine Learning Platforms. We are the only provider in the market to offer both Predictive AI and Generative AI on premise and air gapped, in addition to supporting all cloud environments

About This Opportunity 

H2O.ai is at the forefront of the rapidly evolving MLOps landscape. We're leveraging MLOps to transform machine learning models from isolated, engineer-specific tools into robust, cloud-native services that are scalable and consistently available. Our approach is firmly rooted in Kubernetes, positioning our team at the cutting edge of cloud solutions.

We are seeking a highly skilled and motivated Senior Machine Learning Engineer to join our team in Australia. The successful candidate will have a strong background in machine learning and software engineering, with experience in developing and deploying machine learning models in real-world applications. The ideal candidate will have a passion for working with data, a strong understanding of machine learning algorithms and techniques, cloud technology and the ability to communicate complex technical concepts to stakeholders.

This position is based in Sri Lanka.

What You Will Do 

As a Senior Machine Learning Engineer, you will work closely with technical teams on the customer side and Customer Success, Enterprise Support, and Product Engineering teams on the H2O side. 

Your primary responsibilities are:

• Deliver technical professional services to the customer.
• Working closely with H2O Data Scientists in advising and developing end to end machine learning solutions for the customer requirements.
• Integrating H2O products with customer data sources for model training.
• Integrating machine learning models/pipelines (python and mojo scoring pipelines) with customer systems for scoring (relatime/batch) as well as model monitoring and operations.
• Implementing end to end MLdata flow pipelines that help streamline and data science solutions to a business problem.
• Implementing AI driven applications using the open source H2O Wave SDK.
• Provide/gather customer feedback so that you can work with the H2O.ai Engineering team to further enhance our products for needed features.
• Ensure the scalability, reliability, and performance of deployed models by implementing appropriate monitoring, testing, and debugging processes.
• Work on GenAI tools - LLMs, Agents, Fine tuning engines, RAG, create GenAI Apps

Your additional responsibilities include:

• Be the trusted solutions advisor for our customers and partners.
• Communicate effectively with a diverse audience of internal and external stakeholders consisting of: Engineers, business people, partners, executives.
• Translate business cases and requirements into value based technical solutions through the architecture of machine learning workflows and systems from data ingestion to model deployment.

What We Are Looking For 

• Bachelor’s or a higher education degree in Computer Science/Engineering, data science, statistics or related field

Data Engineering Skills

• Experience building data pipelines, ETL data sets, preferably on ‘Big Data’
• Excellent understanding and experience with big data tools like Hadoop and Spark
• Excellent knowledge of SQL query language and working with relational databases.
• Understanding of various NoSQL database types and their application scenarios
• Experience in Spark/Kafka and Hadoop ecosystem

Programming Languages/Frameworks

• Proficient in Python or R for data science. Java, Bash scripting, Scala Go are a plus
• Experience with writing REST API using microservices frameworks in Python or Java
• Experience in building and deploying agentic/RAG AI applications
• Experience with dockerization of services (i.e. creating docker images)
• Understanding of Kubernetes based application development

Data Science skills

• Experience of visualizing and presenting (EDA) to stakeholders using H2O Wave (plus), or other standard data visualization libraries in the Python and R stacks or using Tableau/PowerBi.
• Experience with post production model monitoring tools like H2O MLOps (plus) MLFlow etc.
• Understanding of using a variety of machine learning techniques (supervised/unsupervised, clustering,
• decision tree learning, neural networks, etc.) and their real-world advantages/drawbacks/tuning
• techniques.
• Understanding of using advanced statistical techniques and concepts (regression, properties of distributions, statistical tests and proper usage, etc.) for practical applications.

Additional Requirements

• Experience of working in a customer-facing environment, providing technical services
• Excellent communication skills (proficient in spoken and written English). Additional languages are a plus.
• Amicable attitude. Aptitude to independently investigate and find solutions to technical problems; urge to learn/master new technologies. Maker mindset.

How to Stand Out From the Crowd 

• Experience with big data technologies such as Hadoop, Spark, or NoSQL databases.
• Experience in designing & building multi-agents frameworks and solutions 
• Familiarity with DevOps practices and tools such as Git, Jenkins, ArgoCD.
• Knowledge of data privacy and security principles.
• Experience with natural language processing or computer vision tasks.
• Experience with model interpretability techniques such as feature importance, partial dependence plots, or SHAP values.
• Experience with IaC technologies such as Terraform, AWS Cloud Formation.
• Experience with containerization technologies such as Kubernetes, Docker.
• Experience of debugging/troubleshooting containerized workloads and familiarity with tools such as kubectl, helm etc. 

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 legally protected basis.

H2O.ai is an innovative AI cloud platform company, leading the mission to democratize AI for everyone. Thousands of organizations from all over the world have used our cutting-edge technology across a variety of 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 responsible AI drive the company’s purpose.

Please visit www.H2O.ai to learn more.
#LI-Hybrid