Senior AI Engineer , Physics AI
Firmus Technologies
Firmus Technologies is a global leader pioneering the development and operation of efficient AI infrastructure across Asia Pacific.
Founded in Australia in 2019, our mission is to create the most efficient AI infrastructure by combining cutting-edge technology with a steadfast commitment to sustainability.
At Firmus, we are unique in our approach. We design, build, and operate a new class of digital infrastructure – the AI Factory. Through our model-to-grid technology approach, we have pushed the boundaries of multi-generational liquid cooling systems, energy management, AI software orchestration, and construction. For our customers, this approach allows us to make every watt count and deliver low-cost AI tokens globally.
Firmus AI Cloud
Our large-scale GPU cloud platform, Firmus AI Cloud, is purpose-built to deliver energy-efficient AI compute at scale to customers.
It empowers developers, enterprises, educational institutions, and government users to train and deploy AI models with unmatched efficiency and cost savings. With an ever-growing suite of services and applications, we are committed to delivering a cloud experience that is market-leading, proprietary, and built to scale.
Why Firmus?
As an NVIDIA Cloud and Engineering partner in Asia Pacific, you will gain skills, experience, and exposure across the AI industry and be part of shaping what this industry looks like for decades to come.
We are founder-led, not a big corporate. Decisions happen fast, our leaders are accessible, and there's minimum bureaucracy between you and the work. Ownership comes early. Whatever your role, you will have a direct line to outcomes, helping shape how the business grows as we scale nationally across a long-term, large-scale roadmap.
Work alongside founders and experts in AI infrastructure, energy systems and next-generation compute.
What we build here has impact beyond the business. Our AI Factories are designed to operate as assets to the energy grid to actively strengthen the communities and regions they operate in rather than drawing from them.
Considering applying? You don't need a perfect background to join our team. If you're driven and curious, there's a path for you. We back our people to grow into new domains and take on challenges beyond their previous experience.
ROLE SUMMARY
The Senior AI Engineer, Physics AI will build Firmus’s physics-AI capability, turning authoritative engineering inputs into qualified surrogate models and production software that accelerate AI Factory design, optimisation, simulation and operations. You will connect high-fidelity engineering analysis with fast, inspectable decision support across compute, cooling, power and infrastructure systems.
This is a hands-on engineering role at the intersection of scientific machine learning, numerical simulation and software delivery. Working with CFD, mechanical, electrical, controls, network, AI infrastructure and operations specialists, you will own the learned-model lifecycle and the services that make it useful. Engineering specialists retain authority over physical assumptions, solver evidence, design acceptance and protection systems.
KEY RESPONSIBILITIES
- Scientific machine learning and surrogate models
Develop and qualify surrogate models for bounded thermal, airflow, cooling and infrastructure scenarios. Start with a well-defined simulation-backed use case, selecting convolutional models, neural operators or graph-based approaches according to the data representation, fidelity and latency required. Extend to geometry-generalising approaches only when data coverage and validation justify them.
- Engineering data and evidence
Build reproducible pipelines from Ansys Fluent and other engineering tools into model-ready datasets. Preserve geometry, mesh, equipment, boundary-condition, units and solver-version lineage, and define canonical field and graph representations with the relevant engineering owners in Engineering and Technology. Design evaluation splits that prevent leakage across related cases and test unseen operating conditions and configurations. Treat approved solver studies and measured evidence as reference inputs, not as data to be stripped of engineering context.
- Production services and integration
Deliver tested, versioned software rather than isolated notebooks. Build training pipelines, model registry and qualification workflows, inference APIs and scenario-orchestration integrations. Own model packaging, qualification workflows, automated testing, monitoring and technical documentation in partnership with platform engineering. Deploy surrogate model endpoints through the AI & Applications inferencing platform and execute on shared GPU capacity managed by the AI & Applications scheduler. Connect model outputs to the engineering optimiser, evidence-linked design records and visualisation services, including Omniverse and OpenUSD where appropriate.
- Qualification, uncertainty and safe use
Agree acceptance criteria with engineering owners before a model is released. Evaluate field accuracy, critical local errors, physical consistency, inference latency and performance against simpler baselines. Implement applicability-domain checks, calibrated uncertainty where supported, out-of-distribution detection and clear rejection or solver-fallback behaviour. Make every result traceable to its inputs, model version, validation evidence and limitations. Keep operational evaluation in shadow mode unless separately approved; surrogate services must not bypass independent controls or protection systems.
- Whole-factory decision support
Integrate surrogates into scenario screening, design comparisons, commissioning rehearsal and operating-envelope studies. Work with engineering teams to assess delivered compute performance, whole-factory energy efficiency, auxiliary loads, cost and risk within explicit system boundaries. Introduce workload-intent and model-to-grid scenarios as the evidence base matures. Prioritise improvements that shorten engineering decision cycles without weakening assurance, and build reusable capability across factory configurations rather than one-off demonstrations.
SKILLS AND EXPERIENCE
You bring deep, hands-on experience in scientific machine learning, computational engineering or simulation-led software, with a track record of taking models from research into dependable production services across multiple projects. You have technically led at least one cross-disciplinary programme spanning ML, simulation and software delivery. You are strong in Python and PyTorch, and can develop, train, profile, evaluate and deploy models on GPU infrastructure. Your experience includes surrogate modelling, neural operators, graph neural networks, reduced-order modelling, physics-informed learning or closely related methods.
You are comfortable working with simulation data, meshes, geometry, boundary conditions and numerical validation. You understand how discretisation, convergence, conservation, data coverage and measurement uncertainty affect a learned approximation. You can distinguish interpolation from unsupported extrapolation and explain when a surrogate should defer to an authoritative engineering calculation.
You have built maintainable software with APIs, containers, version control, automated tests, CI/CD and observability. You can design reproducible data and model contracts, diagnose failure modes and work constructively across engineering disciplines. A degree in engineering, computer science, applied mathematics, computational physics or a related discipline, or equivalent practical depth, provides a relevant foundation.
PREFERRED EXPERIENCE
Experience with NVIDIA PhysicsNeMo or Modulus, PyTorch Geometric, graph or operator learning, uncertainty quantification and multi-GPU training is highly regarded. Familiarity with Ansys Fluent or PyFluent, OpenFOAM, ETAP or comp