Deskripsi Pekerjaan
Are you passionate about pushing the boundaries of artificial intelligence on the world's most powerful computing platforms? Join the elite Frontier team at the National Supercomputing Centre (NSCC) Singapore under the Agency for Science, Technology and Research (A*STAR). We are seeking a highly skilled and innovative HPC AI Engineer to drive the next generation of national-scale AI initiatives. You will play a pivotal role in optimizing massive AI workloads, ensuring our cutting-edge supercomputing infrastructure delivers peak performance for critical scientific research and industrial applications.
In this role, you will bridge the gap between advanced machine learning research and high-performance computing systems. You will work in a collaborative environment alongside world-class researchers and engineers, tackling complex challenges related to data throughput, memory management, and parallel processing. If you are driven by the opportunity to accelerate the future of AI and contribute to a legacy of technological excellence, we want to hear from you.
Tanggung Jawab
- Design, develop, and optimize high-performance computing (HPC) software solutions for large-scale AI training and inference workloads.
- Analyze and profile deep learning frameworks (e.g., PyTorch, TensorFlow) to identify bottlenecks and implement performance improvements.
- Collaborate closely with computational scientists to accelerate complex simulations and AI model deployments on supercomputing clusters.
- Develop and maintain tools for automated benchmarking, profiling, and monitoring of system performance.
- Work with hardware architects to leverage the full potential of heterogeneous computing architectures (GPUs, TPUs, FPGAs).
- Implement scalable distributed training strategies to handle massive datasets efficiently.
- Document technical architecture and provide guidance on best practices for HPC in AI workflows.
Kualifikasi
- Master’s or PhD in Computer Science, Computational Mathematics, Electrical Engineering, or a related technical field.
- Strong proficiency in programming languages such as C++, CUDA, Python, and GPU programming.
- Proven experience in HPC environments, including MPI, OpenMP, and parallel programming concepts.
- Demonstrated experience in optimizing machine learning models for performance on high-performance hardware.
- Deep understanding of deep learning architectures and training methodologies.
- Excellent problem-solving skills and the ability to troubleshoot complex system-level issues.
- Strong communication skills and the ability to work effectively in a fast-paced, multidisciplinary team.