Machine Learning Engineer – Distributed AI & GPU Systems
Westbury Partners Sydney, AustraliaMachine Learning Engineer – Distributed AI & GPU Systems
Westbury Partners Sydney, 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 toimprove, extend, and optimise the technology itself .
About You:
- You’re an experienced Machine Learning Engineer with5+ years of experience focused on machine learning training systems, inference systems, or ML infrastructure.
- You have strong software engineering capabilities inPython, CUDA, or C++ , along with experience using modern frameworks such asPyTorch, TensorFlow, or JAX .
- You understand GPU programming and acceleration technologies such asCuDNN and TensorRT , and ideally have experience with distributed training technologies includingHorovod 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
Job ID 8453,8636
More Jobs From Westbury Partners
Westbury Partners
Sydney, Australia
Westbury Partners
Sydney, Australia
Westbury Partners
Sydney, Australia
Westbury Partners
Sydney, Australia
Westbury Partners
Sydney, Australia
Westbury Partners
Sydney, Australia
Westbury Partners
Sydney, Australia
Westbury Partners
Sydney, Australia
Westbury Partners
Sydney, Australia
Boost your career
Find thousands of job opportunities by signing up to eFinancialCareers today.More Jobs Like This
Block, Inc
San Francisco, United States
Amazon Development Center U.S., Inc.
Seattle, United States
Lloyds Bank plc
Chester, United Kingdom