Machine Learning Engineer (Python / PyTorch)

Dijkstrack LTD; South Africa; Remote; ZAR 500,000 - 600,000; Full time

Job summary

A global technology firm in South Africa is hiring for a remote contract position focused on machine learning and deep learning models. The successful candidate will leverage skills in Python and PyTorch while working within a distributed engineering team on an international project. This role offers exposure to advanced ML solutions and MLOps best practices.

Benefits

Technical community support

Remote-first culture

Global exposure

Qualifications

• Strong proficiency in Python for ML development.
• Hands-on experience with PyTorch.
• Solid understanding of machine learning fundamentals.

Responsibilities

• Design, train, and deploy machine learning models using PyTorch.
• Build and maintain ML pipelines and data processing components.
• Analyse datasets and optimise models for accuracy.

Skills

Python

PyTorch

Machine Learning fundamentals

Model architectures

Large datasets handling

Git workflows

Tools

MLflow

SageMaker

Job description

This 12-month remote contract role places you on an international machine learning project through Dijkstrack for one of our global technology partners. You’ll work within a distributed engineering team to design, train, and deploy ML models using modern frameworks like PyTorch. Dijkstrack engineers enjoy access to technical community support, structured delivery processes, and the opportunity to embed within long-term global product teams.

Key Duties & Responsibilities

• Design, train, and deploy machine learning and deep learning models using PyTorch
• Build and maintain ML pipelines, model training workflows, and data processing components
• Work with product and engineering teams to integrate models into production systems
• Analyse datasets, performance metrics, and optimise models for accuracy and efficiency
• Implement MLOps best practices for model versioning, deployment, and monitoring
• Document models, datasets, evaluations, and maintain reproducibility standards
• Participate in code reviews and uphold engineering quality in ML codebases

Essential Skills & Technical Requirements

• Strong proficiency in Python for ML development
• Hands‑on experience with PyTorch (TensorFlow experience also welcome)
• Solid understanding of machine learning fundamentals, model architectures, training loops
• Ability to work with large datasets, ETL pipelines, and model optimisation
• Familiarity with Git workflows, remote collaboration, and agile delivery models

Beneficial / Nice‑to‑Have Experience

• Experience with ML Ops tools like MLflow, Weights & Biases, SageMaker, or similar
• Exposure to cloud‑based ML deployments (AWS, Azure, GCP)
• Knowledge of microservice integration for ML models
• Prior work in AI product teams or international ML research environments

Benefits of Working with Dijkstrack

• Join a network of engineers delivering advanced ML solutions internationally
• Technical community of senior engineers across data, backend, and product domains
• Remote‑first culture, global exposure, and structured technical delivery support