Machine Learning Research & Engineering Specialist

Gramian Consulting Group; México; Remote; MXN 700,000 - 1,100,000; Full time

Job summary

Gramian Consultancy is seeking a highly skilled Machine Learning Research & Engineering Specialist to contribute to an advanced AI training project focused on model development, training and inference systems, numerical computing, and performance optimization.

This is a coding-intensive role requiring hands-on experience with Python, machine learning engineering, and coding agents integrated into development workflows.

Qualifications

• Master's degree or PhD in Computer Science, ML, AI, or related quantitative field.
• Strong Python programming and ML engineering experience.
• Experience with coding agents within Python development workflows.
• Ability to build reproducible implementations and verify correctness through testing.
• Excellent written communication and problem-solving skills.

Responsibilities

• Develop, implement, and validate ML models, training pipelines, inference systems, and supporting infrastructure.
• Implement or modify model components, data pipelines, evaluation systems, and numerical methods using Python.
• Build reproducible programmatic workflows using Python, CLI tools, and ML frameworks.
• Debug numerical, algorithmic, and system-level issues to identify root causes and solutions.
• Optimize model training and inference for memory, speed, and throughput.
• Design and execute tests to verify correctness and performance.
• Review and validate ML engineering tasks, including code and proposed solutions.
• Use coding agents in development workflows to accelerate implementation, debugging, testing, and review.
• Document approaches, decisions, test results, and performance findings.

Skills

Python programming

Machine Learning engineering

Independent work

Testing & debugging

Numerical reasoning

Communication

Education

Master's degree or PhD in Computer Science/ML/AI/Applied Mathematics

Job description

Gramian Consultancy is a boutique consultancy specializing in IT professional services and engineering talent solutions. With a strong background in software engineering and leadership, we help companies build high-performing teams by matching them with professionals who truly fit their needs.

About the Role

We are seeking a highly skilled Machine Learning Research & Engineering Specialist to contribute to an advanced AI training project focused on model development, training and inference systems, numerical computing, and performance optimization. This is a coding-intensive role requiring hands-on experience with Python, machine learning engineering, and coding agents integrated into development workflows. You will build, modify, debug, and validate machine learning implementations, create reproducible training and inference pipelines, and optimize systems for correctness, memory usage, and throughput. The ideal candidate combines strong quantitative foundations with practical coding skills and experience solving complex machine learning engineering problems.

SENIORITY: Experienced / Research-Level Engineer

EDUCATION: Master's degree or PhD in a relevant quantitative discipline

LANGUAGE: Strong written and verbal English

LOCATION: Remote — Worldwide

Key Responsibilities

Develop, implement, and validate machine learning models, training pipelines, inference systems, and supporting infrastructure.

Implement or modify model components, data pipelines, evaluation systems, and numerical methods using Python.

Build reproducible programmatic workflows using Python, command-line tools, and relevant machine learning frameworks.

Debug numerical, algorithmic, and system-level failures to identify root causes and implement reliable solutions.

Optimize model training and inference for memory usage, computational efficiency, throughput, and performance.

Design and execute tests to verify implementation correctness and compliance with objective performance requirements.

Review and validate challenging machine learning engineering tasks, including existing code and proposed solutions.

Use coding agents as part of the development workflow to accelerate implementation, debugging, testing, and code review while validating generated results.

Document technical approaches, implementation decisions, test results, and performance findings clearly.

• Deliver high-quality, reproducible task submissions within project deadlines.

Requirements

Master's degree or PhD in Computer Science, Machine Learning, Artificial Intelligence, Applied Mathematics, Statistics, Engineering, or a closely related quantitative discipline.

Strong professional or research experience in machine learning and practical ML engineering.

Strong Python programming skills, including the ability to implement, debug, test, and optimize non-trivial code.

Hands-on experience using coding agents within a Python development workflow.

Understanding of machine learning model development, training and inference workflows, and evaluation methods.

Ability to build reproducible implementations and verify correctness through systematic testing.

Strong quantitative reasoning and problem-solving skills, particularly when diagnosing numerical or computational issues.

Ability to work independently, manage technical tasks, and meet defined quality and submission requirements.

Strong written communication skills and the ability to explain technical findings.

Availability to contribute approximately 15 hours per week and begin the first task within 24–48 hours after onboarding.

Machine Learning Research & Engineering Specialist MX