AI/ML Engineer

🤖 AI/ML Engineer

Locations: Colombo

About the Role

Axiata Digital Labs is seeking a skilled data science professional on a contract basis to design, build, and optimize advanced machine learning and artificial intelligence solutions.

What You Will Do

• Design, develop, and optimize classical machine learning models including regression, classification, clustering, time-series forecasting, and anomaly detection.
• Build and deploy deep learning models using frameworks such as TensorFlow or PyTorch for structured, unstructured, and multimodal data.
• Fine-tune and evaluate language models (LLMs/SLMs) for tasks like text classification, summarization, and domain-specific reasoning.
• Implement and maintain MLOps and LLMOps pipelines, covering training, versioning, CI/CD, deployment, rollback, and lifecycle management.
• Develop model monitoring and observability solutions addressing performance, drift detection, bias, latency, and cost metrics.
• Apply AIOps concepts to automate detection, root cause analysis, and predictive insights using operational and telemetry data.
• Collaborate with API Manager and platform teams to expose machine learning and AI capabilities as secure, scalable APIs.
• Participate in data preparation, feature engineering, rigorous model validation, experimentation, and benchmarking.
• Mentor junior engineers and interns while contributing to team technical standards and staying updated with emerging AI technologies.

What You Need

• Bachelor's degree in IT, Computer Science, Data Science, Engineering, Mathematics, or a related field.
• 1 to 3 years of hands-on experience in data science or machine learning engineering roles.
• Strong proficiency in Python and common machine learning and deep learning libraries such as scikit-learn, PyTorch, TensorFlow, NumPy, and pandas.
• Proven experience in developing and deploying production-grade ML models and MLOps platforms like MLflow, Kubeflow, SageMaker, or Vertex AI.
• Practical exposure to LLMOps, prompt engineering, fine-tuning, evaluation, and model serving.
• Experience working with APIs, microservices, cloud platforms (AWS, Azure, GCP), and containerization tools like Docker and Kubernetes.
• Familiarity with time-series analysis, vector databases, RAG pipelines, and AIOps monitoring techniques.
• Strong problem-solving skills with a pragmatic, engineering-first mindset and ability to communicate complex concepts clearly.

Work Arrangement

• 6-month contract role
• Located in Colombo, Sri Lanka

How to Apply

Apply through the organisation's careers page via the Apply button.

Contact Information

• Apply: https://www.careers-page.com/axonect/job/W387685W