Machine Learning Engineer

Type: Freelance / Contract

Location: Fully Remote

Hours: Around 20 hours/week, with flexible scheduling

Process: Short screening call → Technical assessment → Onboarding

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.