1,002 Data Science Teams jobs in Kenya

Role Overview:

We are looking for a Senior AI/Machine Learning Engineer (7+ years of experience) who is a stellar individual contributor with recent hands-on expertise in fine-tuning and large-scale training of modern models (LLMs/VLMs).

This is a hands-on role where you will :

• Lead fine-tuning workflows, large-scale training runs, and evaluation design.

• Collaborate closely with researchers to bring cutting-edge approaches from papers into production.

• Work directly with customers to align performance metrics, validation, and deployment readiness.

Note: This is not a managerial role. We are seeking candidates who are currently active ICs, heavily involved in model fine-tuning/training in the last 8–12 months.

What You'll Do Day-to-Day

• Fine-tune and train LLMs/VLMs at scale (LoRA/QLoRA, PEFT methods).

• Build reproducible training pipelines with distributed training and mixed precision.

• Design and run robust evaluation frameworks (task-specific + lm-eval-harness).

• Translate research papers into working implementations, collaborating with researchers.

• Work with customers to validate models against business and performance needs.

• Optimize training runs with profiling, performance tuning, and efficiency improvements.

• Maintain experiment tracking, reproducibility, and structured model artifacts.

Requirements

• 7+ years in ML/AI engineering, with a strong recent focus on fine-tuning and training large models.

• Expert in PyTorch, Hugging Face Transformers, and PEFT (LoRA/QLoRA).

• Strong experience with distributed training (DDP, FSDP, DeepSpeed, Accelerate).

• Skilled with evaluation frameworks (lm-eval-harness, custom metrics, task benchmarks).

• Proven ability to reproduce and improve results from recent research papers.

• Strong coding practices in Python, with modular, clean implementations.

• Familiarity with experiment tracking tools (Weights & Biases, MLflow).

• Ability to interact with customers and researchers to translate requirements into engineering solutions

(Bonus) Preferred Qualifications:

• Experience with other ML libraries (e.g., PyTorch, Flax)
• Background in ML research or scientific computing
• Experience with production model monitoring and governance

Company Description

At AgroNest Ventures Private Limited, we are transforming the future of agriculture with cutting-edge solutions that empower farmers, optimize resources, and promote sustainability. As a forward-thinking agri-tech company, we are committed to delivering technology-driven solutions across the agricultural value chain. We develop advanced agricultural technologies, promote eco-friendly practices, and empower farmers with data-driven insights and modern tools. Our vision is to be a global leader in sustainable agriculture, fostering prosperity through innovation and transforming how the world grows its food.

Role Description

This is a full-time hybrid role for a Machine Learning Engineer located in Nairobi County, Kenya, with some work-from-home flexibility. The Machine Learning Engineer will be responsible for developing and implementing machine learning algorithms and models, conducting research on neural networks, and applying pattern recognition techniques. Day-to-day tasks also include collaborating with cross-functional teams, analyzing datasets, and optimizing machine learning models to improve agricultural practices and outcomes.

Qualifications

• Proficiency in Pattern Recognition and Neural Networks
• Strong background in Computer Science and Algorithms
• Solid knowledge of Statistics
• Experience with machine learning frameworks and tools
• Excellent problem-solving and analytical skills
• Ability to work independently and in a team environment
• Experience in the agriculture industry is a plus
• Bachelor's or Master's degree in Computer Science, Machine Learning, or a related field

• Design, develop, and implement machine learning models and algorithms.
• Collaborate with data scientists to preprocess data, select features, and train models.
• Deploy ML models into production environments, ensuring scalability and reliability.
• Monitor and maintain deployed models, making necessary adjustments and improvements.
• Optimize model performance for speed and efficiency.
• Develop and maintain ML pipelines and infrastructure.
• Stay current with the latest advancements in machine learning and AI research.
• Work with software engineering teams to integrate ML models into applications.
• Conduct experiments to evaluate different modeling approaches.
• Document ML models, methodologies, and results.

• Bachelor's or Master's degree in Computer Science, Engineering, Statistics, or a related quantitative field.
• Proven experience as a Machine Learning Engineer or similar role.
• Proficiency in programming languages such as Python, R, or Java.
• Strong experience with ML libraries and frameworks (e.g., TensorFlow, PyTorch, Scikit-learn).
• Solid understanding of statistical modeling, data mining, and ML algorithms (e.g., regression, classification, clustering, deep learning).
• Experience with data preprocessing, feature engineering, and model evaluation.
• Familiarity with cloud platforms (AWS, Azure, GCP) and MLOps principles.
• Excellent problem-solving and analytical skills.
• Strong communication and teamwork abilities.
• Must be able to work on-site in **Kisumu, Kisumu, KE**.