Software Engineer, Machine Learning, Johannesburg

We are seeking a talented and motivated Mid-Level Software Engineer, specialising primarily in Machine Learning and secondarily in Artificial Intelligence.

The successful candidate will have a strong foundation in building, deploying, and maintaining machine learning models, along with hands-on experience in MLOps practices and modern Python-based ML frameworks.

You will play a key role in designing scalable ML solutions, improving model performance, and contributing to AI-driven product innovation within a collaborative engineering environment.

Key Responsibilities

• Design, develop, and deploy machine learning models for production environments
• Build and maintain scalable ML pipelines and workflows
• Implement and manage MLOps practices, including CI/CD for ML models
• Collaborate with cross-functional teams (engineering, data, and product) to deliver AI-driven solutions
• Optimise model performance, accuracy, and efficiency
• Work with large datasets for data preprocessing, feature engineering, and model training
• Monitor and maintain deployed models to ensure reliability and performance
• Contribute to code reviews, documentation, and engineering best practices

Required Skills & Competencies

• Hands-on experience with PyTorch
• Experience with Python ML libraries, including:
  • scikit-learn
  • pandas & NumPy
  • TensorFlow (advantageous)
• Strong understanding of MLOps practices and tools, such as:
  • Model versioning
  • Experiment tracking
  • CI/CD pipelines for ML
• Experience with data processing and pipeline orchestration tools
• Familiarity with cloud platforms (AWS, Azure, or GCP)
• Understanding of REST APIs and model integration
• Experience with containerisation tools (Docker, Kubernetes)
• Exposure to distributed computing or big data technologies
• Knowledge of model monitoring and observability tools
• Experience working in Agile development environments

Qualifications & Experience

• Bachelor's degree in Computer Science, Engineering, or a related field (or equivalent practical experience)
• 4–5 years of professional software engineering experience
• Strong experience in Machine Learning as a primary focus
• Working knowledge of Artificial Intelligence concepts and applications
• Proficiency in Python and relevant ML libraries

Soft Skills

• Strong problem-solving and analytical thinking skills
• Ability to work independently as well as collaboratively
• Good communication skills, with the ability to explain technical concepts to non-technical stakeholders
• High attention to detail and a strong commitment to quality