About this role
Senior Machine Learning Engineer - Research Optimisation at Canva. Location: Sydney, New South Wales, Australia. Role: productionising models, profiling workloads, optimising inference Requirements: Experienced ML engineer able to productionise research models, profile and optimise PyTorch training/inference, work with Kubernetes and distributed training (FSDP, DDP, DeepSpeed), and collaborate with researchers; strong Python and cloud (AWS) experience. Category: Software Development Seniority: Senior Level Tools: Python, PyTorch, FSDP, DDP, DeepSpeed, Megatron, ONNX, TorchScript, Triton, CUDA, torch.compile, TensorRT, Weka, Vast, Lustre, Kubernetes, AWS Commitment: Full Time Workplace: Hybrid Languages: English