Airwallex - Senior Software Engineer, Machine Learning Platform

• 5+ years of software engineering experience, with at least 3+ years focused on model training infrastructure, model serving systems, or MLOps platforms. • Bachelor's or Master's degree in Computer Science, Engineering, or a related technical field. • Hands-on experience with modern deep learning frameworks (e.g., PyTorch, TensorFlow, JAX) and model training execution engines. • Strong proficiency in core programming languages such as Python, Java, or C++. • Experience with distributed orchestration and workflow management tools (e.g., Kubernetes, Ray, Kubeflow Pipelines, Airflow). • Solid understanding of GPUs, including GPU architecture, hardware acceleration, and GPU-based training or inference optimization. • Experience with model acceleration frameworks and Large Language Models (LLMs). • Proficiency in performance profiling and bottleneck identification using tools like NVIDIA Nsight Systems for training and inference optimization. • Experience with cloud platforms (e.g., AWS, GCP) and building large-scale, low-latency production machine learning infrastructure. • YOU'LL THRIVE HERE IF • You’re comfortable owning the roadmap yourself. • You own the outcome and you don't wait for permission to fix what's broken. • You're comfortable with ambiguity. Give you a problem, not a prescription, and you'll run with it. • You enjoy working closely with people across multiple countries and time zones as part of one connected, global team, including flexing your hours occasionally to make that connection work. • You value in-person collaboration and are happy being in the office five days a week.