AI Engineer

About the Company

H2O.ai, founded in 2012, focuses on making AI broadly accessible. The company is known for combining generative and predictive AI to help businesses and public sector teams create custom GenAI applications on their own data. Its approach emphasizes sovereign AI deployments that are secure, compliant, and adaptable to different infrastructure needs, with strong attention to privacy and data control.

The company’s open-source products are used by more than 20,000 organizations globally, including over half of the Fortune 500. Its technology supports AI programs at organizations such as AT&T, Commonwealth Bank of Australia, Chipotle, Workday, Progressive Insurance, and NIH. H2O.ai also works with partners including NVIDIA, Dell Technologies, Deloitte, Ernst & Young, Snowflake, AWS, Google Cloud Platform, VAST Data, and MinIO. Through its AI for Good initiative, the company supports nonprofits, foundations, and communities working in education, healthcare, and environmental conservation. It also has a community of 2 million data scientists worldwide.

The company has raised 256 million from investors including Commonwealth Bank, NVIDIA, Goldman Sachs, Wells Fargo, Capital One, Nexus Ventures, and New York Life.

Role Overview

This position is for an AI Engineer based in Melbourne, Australia, with a remote-friendly setup. The role is centered on building practical AI systems that solve complex enterprise problems across APAC. You will work across agentic AI systems, LLM-based applications, and production-grade machine learning pipelines, with your output expected to be deployed and used in real customer environments from the outset.

You will collaborate with Kaggle Grandmasters, ML engineers, and subject matter experts to deliver solutions that move beyond prototypes and demonstrators into real operational workflows and measurable business results.

Responsibilities

• Develop, implement, and release AI applications, machine learning models, and agent-based systems using the H2O AI platform.
• Handle the full ML lifecycle, including data preparation, feature creation, model training, evaluation, deployment, and monitoring.
• Create production-ready AI capabilities that combine predictive ML, generative AI, retrieval-augmented generation, and LLM-driven components.
• Build backend services and APIs that connect AI models with data sources and application layers.
• Help deploy AI solutions in client environments for both predictive and GenAI use cases.
• Participate in technical workshops, product demos, and training or enablement sessions.
• Work with senior engineers and cross-functional teams to create reusable modules and reference architectures.
• Explain technical work clearly to both technical teams and non-technical stakeholders.

Requirements

• 2 to 5 years of experience in AI, machine learning, data science, or software engineering.
• Solid grasp of predictive machine learning, including feature engineering, model training, evaluation, and interpretation.
• Hands-on experience deploying ML or AI solutions in production or production-like settings.
• Familiarity with generative AI concepts such as LLMs, SLMs, RAG, prompt engineering, or agentic systems.
• Strong Python skills and sound software engineering practices, including testing, debugging, version control, and CI/CD.
• Experience creating APIs or backend services.
• Basic knowledge of cloud environments and MLOps or LLOps concepts.
• Good communication and collaboration skills.

Preferred Profile

• Experience using H2O products.
• Background in full-stack or application development.
• Prior hackathon, internship, or open-source involvement.
• Understanding of model governance or enterprise AI processes.
• Exposure to enterprise or regulated data.

Why Join

• Competitive total rewards position.
• Remote-friendly working culture.
• Flexible work arrangements.
• Opportunity to work with a highly skilled team.
• Strong scope for career growth.

Equal Opportunity and Additional Information

H2O.ai promotes a diverse and inclusive workplace. All qualified applicants are considered without discrimination based on race, ethnicity, religion, gender, sexual orientation, age, disability, or any other protected characteristic.

The company also reiterates its mission to democratize AI for everyone through innovative cloud-based technology used across many industries. It emphasizes its work with top machine learning practitioners and its commitment to responsible AI and AI for Good.

Cookie and analytics consent information was included in the source and does not affect the job details.

Location

This role is based in Melbourne, Victoria, Australia.