Remote GPU Software Engineer – CUDA, AI & ML Surrogates

Bmt Defence Services LTD; City of Brisbane; Hybrid; AUD 120,000 - 180,000; Full time

Benefits offered by this job

Remote work

Hybrid work

Part-time

Defence Reserve Leave

Employee Assistance Program

Learning and development

Job summary

TUFLOW in Brisbane is seeking a Software Engineer to join our team developing hydraulic modelling software. The role focuses on CUDA C/C++ solvers, performance improvements, and integrating AI-assisted development into the workflow.

You will work with a mature codebase written in Fortran, CUDA C++, Python, and Rust, contribute to GPU compute and AI applications, and collaborate with engineers and domain specialists to deliver accurate and efficient simulations.

Qualifications

• A bachelor’s degree in software engineering, computer science or another relevant discipline.
• 5+ years of software development experience.
• Experience with version control, code review, automated testing and issue tracking.
• Experience with C/C++ and CUDA development or other GPU compute platforms.
• Baseline AI-assisted development proficiency with coding assistants and agentic tooling.
• Experience with AI/LLM tooling and architectures (RAG, MCP).
• Experience developing or integrating machine learning surrogate models.
• Experience in flood or coastal engineering or numerical scientific modelling domains.

Responsibilities

• Undertake complex technical and software development tasks, exercising professional judgement, initiative and appropriate autonomy.
• Plan and prioritise work to deliver agreed outcomes within scope, schedule, budget and quality requirements.
• Develop, test, verify, document and maintain software features across the TUFLOW Software Suite, with a primary focus on GPU compute and AI applications.
• Contribute to the planned enhancement and modernisation of the TUFLOW codebase to support product quality, maintainability, performance and the future needs of users.
• Collaborate with the wider TUFLOW team and other BMT specialists to deliver multidisciplinary development, research and technical initiatives.
• Engage with users and stakeholders, as appropriate, to understand technical requirements, support service delivery and inform software development priorities.
• Work in accordance with applicable quality, safety, environmental, information security and software development procedures and standards.
• Contribute actively to improvement of the TUFLOW software development process, whilst contributing to established processes and procedures.
• Maintain and develop professional and technical capabilities relevant to the role, recognising that it is a broad and rapidly developing domain.
• Work is reviewed at the completion of tasks, and ongoing, usually on a fortnightly basis. Performance is reviewed on a yearly basis, as part of BMT’s global performance review cycle.

Skills

CUDA C/C++

GPU compute

C/C++ development

AI-assisted development

Code review

NVIDIA tooling

Education

Bachelor's degree in CS/SE

Tools

Git

CI/CD

Testing frameworks

Job description

TUFLOW in Brisbane is seeking a Software Engineer to join our team developing hydraulic modelling software. The role focuses on CUDA C/C++ solvers, performance improvements, and integrating AI-assisted development into the workflow.

You will work with a mature codebase written in Fortran, CUDA C++, Python, and Rust, contribute to GPU compute and AI applications, and collaborate with engineers and domain specialists to deliver accurate and efficient simulations.