Westbury Partners – Sydney NSW

Full time

6d ago , from E-Financial Careers Australia

Build high-performance machine learning infrastructure for large-scale training and real-time inference, leveraging distributed computing, GPU acceleration, advanced frameworks, and scalable production systems.

What You'll Do:

- You’ll build the systems that power the training and deployment of sophisticated machine learning models at scale. Working alongside researchers, hardware specialists, and software engineers, you’ll tackle challenging problems across distributed computing, GPU acceleration, model performance, and real-time inference.
- Your work will accelerate experimentation, improve training efficiency, and enable high-performance models to deliver accurate predictions within demanding production environments.

Your responsibilities will include:

- Develop large-scale distributed training pipelines for complex datasets and machine learning models.
- Build and optimise low-latency inference pipelines for real-time production predictions.
- Develop high-performance libraries that enhance machine learning frameworks.
- Maximise training and inference performance through GPU hardware and acceleration technologies.
- Design scalable model frameworks capable of processing high-volume datasets.
- Automate machine learning experimentation, hyperparameter tuning, and model retraining.
- Partner with high-performance computing specialists to optimise workflows and reduce training costs.
- Evaluate and deploy third-party tools that improve model development, training, and inference.
- Investigate the internals of open-source ML frameworks and extend their functionality.
- Identify performance bottlenecks and implement solutions to improve scalability and efficiency.




- Collaborate across engineering and research teams to deliver robust production ML systems.

Why Join Us:

- This is an opportunity to work on technically demanding problems at the intersection of machine learning, distributed computing, and high-performance hardware.
- You’ll work with advanced GPU technologies and large-scale infrastructure while contributing to systems that significantly accelerate research and experimentation.
- The role offers the chance to work closely with researchers and infrastructure specialists, explore the internals of leading open-source frameworks, and develop solutions that push the boundaries of ML performance.
- Rather than simply applying existing tools, you’ll have the opportunity to improve, extend, and optimise the technology itself.

About You:

- You’re an experienced Machine Learning Engineer with 5+ years of experience focused on machine learning training systems, inference systems, or ML infrastructure.
- You have strong software engineering capabilities in Python, CUDA, or C++ , along with experience using contemporary frameworks such as PyTorch, TensorFlow, or JAX .
- You understand GPU programming and acceleration technologies such as CuDNN and TensorRT , and ideally have experience with distributed training technologies including Horovod or NCCL .
- Experience with real-time, low-latency ML systems, cloud platforms, orchestration tools, or open-source contributions is highly valued.
- Above all, you’re curious about how systems work beneath the surface, enjoy solving complex performance challenges, and are motivated to build ML infrastructure that operates efficiently at scale.

#MachineLearning #MLEngineer #ArtificialIntelligence #GPUComputing #CUDA #CPlusPlus #Python #DistributedSystems #DistributedTraining #PyTorch #TensorFlow #JAX #LowLatency #MLInfrastructure #HighPerformanceComputing

#J-18808-Ljbffr

📌 Machine Learning Engineer – Distributed AI & GPU Systems (New South Wales)
🏢 Westbury Partners
📍 New South Wales