Skip to content

Machine Learning Engineer

Key Prediction

Ciudad De México, MXhybrid$40k-$45k/yrPosted Jun 2, 2026

Skills

classificationscikitlearntensorflowairflowpytorchpandaspythonflaskazurenumpycicdgooglecloudaws

About the role

About the Role

We are seeking a highly skilled Senior Machine Learning Engineer to join our growing AI team. In this role, you will design, develop, deploy, and maintain Machine Learning and Artificial Intelligence solutions that drive business value through intelligent automation, predictive analytics, and innovative AI-powered products.

You will work closely with Data Scientists, Data Engineers, Software Engineers, and business stakeholders to transform experimental models into scalable, production-ready solutions.

Key Responsibilities

Design, train, optimize, and deploy Machine Learning models in production environments.

Develop Computer Vision solutions for image and video detection, classification, and segmentation.

Build and maintain scalable data pipelines for model training and inference.

Integrate Machine Learning models through APIs and microservices.

Select, evaluate, and adapt Large Language Models (LLMs) for business use cases.

Implement Prompt Engineering, Retrieval-Augmented Generation (RAG), and fine-tuning techniques.

Develop Generative AI solutions, including intelligent assistants, automated summarization, and recommendation systems.

Monitor model performance, drift, and retraining processes.

Optimize cloud infrastructure costs and model performance across AWS, Azure, or GCP environments.

Collaborate within Agile/Scrum teams to deliver high-quality AI solutions.

Ensure best practices in software development, scalability, maintainability, and documentation.

Partner with cross-functional teams to translate business requirements into AI-driven solutions.

Required Qualifications

Bachelor’s degree in Computer Science, Data Science, Artificial Intelligence, Software Engineering, or a related field.

Master’s degree or postgraduate specialization in Machine Learning, Artificial Intelligence, or related disciplines is highly preferred.

Professional Experience

5+ years of proven experience as a Machine Learning Engineer.

Hands-on experience developing and deploying Machine Learning solutions in production environments.

Strong background in Computer Vision projects.

Experience building scalable AI systems in cloud environments.

Experience working in Agile/Scrum environments.

Technical Skills

Machine Learning & Data Science

Advanced proficiency in Python.

Strong experience with NumPy, Pandas, and related data processing libraries.

Expertise in supervised and unsupervised learning techniques.

Experience with TensorFlow, PyTorch, and Scikit-learn.

Strong understanding of model training, evaluation, optimization, and deployment.

Computer Vision

Experience with OpenCV, Pillow, Albumentations, or similar frameworks.

Knowledge of image classification, object detection, segmentation, and video analytics.

Software Engineering

Experience developing APIs using FastAPI and/or Flask.

Proficiency with Git and version control best practices.

Strong knowledge of SQL and NoSQL databases.

Experience building scalable microservices architectures.

Cloud & MLOps

Experience deploying Machine Learning solutions on AWS, Azure, or Google Cloud Platform.

Familiarity with MLflow, Kubeflow, Airflow, Vertex AI Pipelines, and/or AWS SageMaker.

Understanding of CI/CD practices for Machine Learning systems.

Generative AI & LLMs

Hands-on experience with LLMs such as GPT, Llama, Mistral, Claude, Falcon, or Gemini.

Experience with Prompt Engineering and Retrieval-Augmented Generation (RAG).

Knowledge of fine-tuning techniques and model optimization strategies.

Familiarity with LoRA, quantization, distillation, and other cost-optimization approaches.

Experience building AI-powered assistants, chatbots, and multimodal applications.

Preferred Certifications

AWS Certified Machine Learning – Specialty

Microsoft Certified: Azure Data Scientist Associate

Google Cloud Professional Machine Learning Engineer

Sueldo: $40,000.00 - $45,000.00 al mes

Beneficios:

Estacionamiento de la empresa

Lugar de trabajo: remoto híbrido en Ciudad de México

Compensation

This Machine Learning Engineer role pays $40k-$45k/yr. Within typical range for machine learning engineer roles in Mexico.

Questions about this role

  • How do I apply to this Machine Learning Engineer role at Key Prediction?

    Click "Apply with AI Applyd" above. We auto-fill the application from your resume and answer screening questions in seconds. No copy and paste, no juggling tabs.

  • What's the typical salary for Machine Learning Engineer in Mexico?

    Compensation for Machine Learning Engineer roles in Mexico varies widely by seniority, employer size, and remote vs onsite arrangement. Check the salary range on this listing when published, or browse our Machine Learning Engineer hub for Mexico medians across recent openings.

  • How fast does AI Applyd auto-apply?

    Most applications complete in under 90 seconds. You can track the status in your dashboard and watch the screenshot proof land the moment the application submits.

  • What ATS does Key Prediction use?

    AI Applyd supports Greenhouse, Lever, Ashby, Workday, iCIMS, SmartRecruiters, LinkedIn Easy Apply, and most other ATS platforms. If we can submit through the platform, we do.

Want AI Applyd to auto-apply to roles like this?

We tailor your resume per posting, fill the forms, and track replies for you.