Senior Deep Learning Architect - AI Infrastructure

Placements24; South Africa; On-site; ZAR 1,800,000 - 3,200,000; Full time

Benefits offered by this job

Competitive salary

Bonus opportunities

Comprehensive health and retirement

Paid time off

Job summary

Placements24 seeks a Senior Deep Learning Architect to lead the design and optimization of high-performance AI infrastructure in Pretoria. You will shape scalable pipelines, cluster management, and resource scheduling to empower cutting-edge models.

The role requires deep expertise in DL frameworks, container orchestration, and cloud services across AWS/Azure/GCP. You will drive CI/CD for AI workloads and ensure security and reliability across platforms.

Qualifications

• Master's or PhD in CS/EE with focus on systems or AI.
• 6+ years building large-scale distributed systems; 3+ years AI/ML infra.
• Expertise in TensorFlow, PyTorch, or Keras.
• Experience with Docker, Kubernetes and orchestration.
• Experience with AWS, Azure, or GCP for ML workloads.
• Strong Python, Go, or C++ programming.
• Knowledge of networking, storage, compute architectures.
• Excellent problem-solving and teamwork.
• ML monitoring/observability is a plus.

Responsibilities

• Design scalable DL infrastructure, including cluster management, job scheduling, and resource allocation.
• Optimize training and inference pipelines on GPUs/TPUs using state-of-the-art hardware.
• Develop and maintain CI/CD pipelines for AI models and infra components.
• Collaborate with researchers to understand infrastructure needs.
• Evaluate and integrate new technologies and tools to enhance DL capabilities.
• Troubleshoot and resolve performance and reliability issues within AI infra.
• Develop best practices and documentation for AI infrastructure management.
• Assist in hardware/software resource selection and procurement for AI workloads.
• Ensure security and compliance of the AI infrastructure.

Skills

TensorFlow

PyTorch

Keras

Python

Go

C++

Distributed systems

Cloud platforms

AI/ML infra

Problem-solving

Teamwork

ML monitoring

Education

Master's degree in CS/EE

PhD in CS/EE

Tools

Docker

Kubernetes

Job description

About the Role

Our client, a prominent AI research and development firm, is seeking a highly skilled Senior Deep Learning Architect to join their advanced infrastructure team in Pretoria . This role is critical for designing, building, and optimizing the high-performance computing infrastructure that powers our cutting-edge deep learning models. You will be responsible for ensuring the scalability, efficiency, and reliability of our AI development environments. The ideal candidate will have a profound understanding of deep learning frameworks, distributed systems, and cloud computing. This is an exceptional opportunity to work at the intersection of AI and infrastructure, contributing to breakthroughs in the field.

Key Responsibilities

• Design and implement scalable, robust, and efficient deep learning infrastructure, including cluster management, job scheduling, and resource allocation.
• Optimize deep learning training and inference pipelines for maximum performance using state-of-the-art hardware (GPUs, TPUs) and software techniques.
• Develop and maintain CI/CD pipelines for AI models and infrastructure components.
• Collaborate with AI researchers and engineers to understand their infrastructure needs and provide tailored solutions.
• Evaluate and integrate new technologies and tools to enhance our deep learning capabilities.
• Troubleshoot and resolve complex performance and reliability issues within the AI infrastructure.
• Develop best practices and documentation for AI infrastructure management.
• Contribute to the selection and procurement of hardware and software resources for AI workloads.
• Ensure the security and compliance of the AI infrastructure.

Requirements

• Master's degree or Ph.D. in Computer Science, Electrical Engineering, or a related field with a focus on systems or AI.
• Minimum of 6 years of experience in building and managing large-scale distributed systems, with at least 3 years focused on AI/ML infrastructure.
• Deep expertise in deep learning frameworks such as TensorFlow, PyTorch, or Keras.
• Proficiency in containerization technologies (Docker, Kubernetes) and orchestration tools.
• Strong experience with cloud platforms (AWS, Azure, GCP) and their relevant services for ML workloads.
• Expertise in performance tuning and optimization for high-performance computing environments.
• Proficiency in programming languages such as Python, Go, or C++.
• Solid understanding of networking, storage, and compute architectures.
• Excellent problem-solving and analytical skills.
• Strong communication and teamwork abilities.
• Experience with ML monitoring and observability tools is a plus.

Benefits

• Our client offers a highly competitive salary and benefits package, including performance-based bonuses, comprehensive health and retirement plans, and generous paid time off.
• Employees benefit from continuous learning opportunities, access to state-of-the-art technology, and support for attending relevant industry conferences.
• The company culture in Pretoria emphasizes innovation, collaboration, and professional growth.
• Join our client and play a crucial role in advancing the field of artificial intelligence.