Machine Learning Expert - Remote

YO AI Labs; New York (NY); Remote; USD 83,000 - 165,000; Full time

Job summary

YO AI Labs is seeking highly skilled Machine Learning Engineers to contribute to an AI training project focused on model development, training and inference systems, numerical computing, performance optimization, and Python-based engineering.

This is a highly coding-intensive role requiring hands-on experience with Python and coding agents as part of the development workflow. The role is fully remote, global, with flexible hours and weekly tasks.

Qualifications

• Master's or PhD in a quantitative field.
• Strong ML experience and Python proficiency.
• Hands-on with one or more ML frameworks listed above.

Responsibilities

• Develop, implement, and validate ML models, training pipelines, and inference systems.
• Build model components, data pipelines, and evaluation frameworks.
• Create reproducible workflows using Python and CLI tools.
• Debug numerical, performance, and system-level issues.
• Optimize memory, throughput, and overall model performance.
• Review ML implementations against defined correctness and performance requirements.

Skills

Python

Machine Learning

PyTorch

JAX

NumPy/SciPy

coding agents

Education

Master's degree or PhD in CS/ML/AI/Applied Mathematics/Statistics/Engineering

Tools

PyTorch

JAX

NumPy

SciPy

SGLang

vLLM

llama.cpp

Hugging Face Transformers

Hugging Face Tokenizers

Job description

Machine Learning Engineer

Location: Global, Fully Remote
Job Type: Contractor (~15 hours per week)
Schedule: Flexible - choose your preferred hours and days, including weekends.

We are looking for highly skilled Machine Learning Engineers to contribute to an AI training project focused on model development, training and inference systems, numerical computing, performance optimization, and Python-based engineering.

This is a highly coding-intensive role. Candidates must have hands-on experience using Python and coding agents as part of their development workflow.

What You'll Work On

• Develop, implement, and validate machine learning models, training pipelines, and inference systems.
• Build model components, data pipelines, evaluation frameworks, and numerical methods.
• Create reproducible workflows using Python and command-line tools.
• Debug numerical, performance, and system-level issues.
• Optimize memory usage, throughput, and overall model performance.
• Review and validate machine learning implementations against defined correctness and performance requirements.

Required Qualifications

• Master's degree or PhD in Computer Science, Machine Learning, Artificial Intelligence, Applied Mathematics, Statistics, Engineering, or a closely related quantitative field.
• Strong professional or research experience in machine learning.
• Advanced practical proficiency in Python and coding agents.
• Experience with one or more of the following:
  • PyTorch
  • JAX
  • NumPy / SciPy
  • SGLang
  • vLLM
  • llama.cpp
  • Hugging Face Transformers
  • Hugging Face Tokenizers
• Equivalent tools may also be considered where candidates demonstrate relevant technical depth.
• Experience at a technology company, AI laboratory, research organization, or recognized engineering environment is preferred.
• Strong open-source or academic experience may also qualify.

Compensation

Compensation is output-based, with payment provided per task that meets project specifications. Task completion time may vary depending on experience and workflow. Minimum submission requirements apply.

Start Timeline & Availability

We typically fill roles within 48 hours and are seeking experts who can start immediately. Selected candidates should begin their first task within 24-48 hours of completing onboarding.