ML Engineer - Remote

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

Job summary

YO AI Labs is seeking highly skilled Machine Learning Engineers for a fully remote contract role. You will develop models, training pipelines, and inference systems, focusing on Python-based engineering and numerical methods.

The role is highly coding-intensive with hands-on experience using Python and coding agents, and involves optimizing performance, memory usage, and system-level behavior across scalable ML tasks.

Qualifications

• Master's degree or PhD in Computer Science, ML, AI, Applied Mathematics, Statistics, Engineering or related quantitative field.
• Strong professional or research experience in machine learning.
• Advanced practical proficiency in Python and coding agents.
• Experience with PyTorch, JAX, NumPy/SciPy, and one or more ML tooling frameworks.

Responsibilities

• 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.

Skills

Python proficiency

ML background

Coding agents experience

Education

Master's degree or PhD in Computer Science/ML/AI/Applied Math/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.