Company description:

At Komatsu Australia you'll be part of an inclusive culture where all our people thrive, reflected in our purpose: creating value together. We encourage people with diverse backgrounds to apply for this role.

Job description:

The Opportunity

Join Komatsu in a cutting-edge role as an AI Engineer! You will develop and maintain AI and ML models and applications using Databricks, AI/ML libraries and LLMs, collaborating with cross-functional teams to enhance our AI systems. Your work will focus on integrating solutions within Komatsu's Azure and Databricks environments, improving model accuracy and performance.

You will play a key role in advancing our AI capabilities, including large language models (LLMs) and agentic AI systems, contributing to both internal efficiency and customer-facing solutions. Collaborating closely with cross-functional teams, you will help shape the future of AI-driven technology at Komatsu.

Role Located: Fairfield, NSW

Remuneration: $140,00 pa + 5% bonus + super

Komatsu benefits:

· Families Policy: 16 weeks of paid primary carer parental leave, 4 weeks of secondary carer leave, plus many more family friendly benefits

· Financial: Income Protection Insurance provided through our company superannuation fund.

· Career: Recognition, awards, flexibility, further education studies and internal career development

· Health: Corporate discount with Bupa, Total Wellbeing Solution with Telus Health

· Travel: Discounts with Avis, Qantas, Crown, IHG and travel agents

· Yearly Bonus: Potential to receive a yearly bonus

· Referral Program: $3000 referral payment for referring a successful new employee to Komatsu

What we are looking for:

· Proven experience with Databricks platform, including at least two years of hands-on work

· Strong programming skills in Python, including proficiency with Pandas and PySpark libraries

· Skilled in end-to-end ML predictive model development — from data preprocessing and feature engineering to model training, evaluation, and deployment using tools such as Python, scikit-learn, and Databricks.

· Minimum three to four years of experience working with Large Language Models (LLMs), including fine-tuning and RAG architecture

· Ability to independently develop and deploy AI models using modern AI concepts and tools

· Familiarity with containerized deployment environments and integration with Azure and Databricks platforms

· Strong analytical skills with a focus on achieving high accuracy in AI outcomes

· Effective communication skills to collaborate with cross-functional teams, including business units and data teams

· Ability to adapt to rapidly evolving AI technologies and frameworks, contributing to architectural decisions in collaboration with the team

· Commitment to delivering high-quality, scalable AI solutions aligned with business needs and feedback

Ready to get started? Simply hit the apply now button!

Komatsu offers a zero-harm culture in the workplace and competitive remuneration package. We employ people with different skills, abilities, cultural backgrounds, experiences, preferences, ethnicities and from different generations. Komatsu is committed to a workplace where women can thrive.