Senior Deep Learning Architect - AI Infrastructure

Placements24; Pretoria; Hybrid; ZAR 1,800,000 - 2,400,000; Full time

Benefits offered by this job

Bonus

Health insurance

Retirement plan

Generous paid time off

Learning opportunities

Conference attendance

Career growth in Pretoria

Job summary

Placements24 is seeking a Senior Deep Learning Architect to lead the design and optimization of scalable AI infrastructure in Pretoria. You will shape cluster management, job scheduling and resource allocation to support cutting-edge DL models.

You will optimize training/inference pipelines on GPUs/TPUs, develop CI/CD for AI models, and collaborate with researchers to deliver secure, compliant, high-performance infra.

Qualifications

• Master's or Ph.D. in CS/EE or related with focus on systems or AI.
• 6+ years building large-scale distributed systems; 3+ years in AI/ML infra.
• Deep expertise with TensorFlow, PyTorch or Keras.
• Proficient in Docker, Kubernetes and cloud ML services.
• Strong programming in Python, Go or C++.

Responsibilities

• Design scalable DL infrastructure including cluster management and resource allocation.
• Optimize training/inference pipelines on GPUs/TPUs for performance.
• Develop CI/CD pipelines for AI models and infra components.
• Collaborate with researchers and engineers to meet infra needs.
• Evaluate new tools and integrate to enhance DL capabilities.
• Troubleshoot performance and reliability issues in AI infra.
• Develop best practices and documentation for AI infra management.
• Participate in hardware/software procurement decisions.
• Ensure security and compliance of the AI infrastructure.

Skills

Distributed systems design

AI infrastructure

Cloud computing

Team collaboration

Problem solving

Communication

Education

Master's degree or Ph.D. in CS/EE or related (systems/AI)

Tools

Docker

Kubernetes

TensorFlow

PyTorch

Keras

AWS

Azure

GCP

Python

Go

C++

Networking

Storage

Compute architectures

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.