Machine Learning Engineer

This is a remote position.

Type: Freelance / Contract

Location: Fully Remote

Hours: Around 20 hours/week, with flexible scheduling

Process: Short screening call → Technical assessment → Onboarding

Requirements

DevFixr is recruiting Machine Learning Engineers for a client building training environments for frontier AI labs. You’ll work on challenging ML engineering tasks designed to train and evaluate advanced AI models.

What You’ll Do

• Design realistic and challenging ML engineering coding tasks based on real-world codebases.
  
• Build working reference solutions for each task.
  
• Create automated tests to verify that solutions are correct and robust.
  
• Ensure tasks are clearly defined and cannot be solved through shortcuts.
  
• Improve tasks based on reviewer feedback.
  

What We’re Looking For

• 4+ years of hands-on ML Engineering experience in Python.
  
• Candidates with 3+ years of relevant Python/ML experience may also be considered when their education and practical experience demonstrate strong technical capability.
  
• Experience training or fine-tuning models using PyTorch, JAX, TensorFlow, or similar frameworks.
  
• Strong Python coding, debugging, testing, and experience working with large or multi-file codebases.
  
• Strong written English and ability to write clear technical specifications.
  

Nice to Have

• GPU/ML systems experience: CUDA, Triton, FSDP, DeepSpeed, kernel optimization, multi-GPU or distributed training.
  
• Experience with AI training/evaluation, benchmarks, model evaluations, or RL environments.
  
• Kaggle, competitive programming, or open-source ML contributions.