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.
Robust Python coding, debugging, testing, and experience working with large or multi-file codebases.
Solid written English and ability to write explicit 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.

📌 Machine Learning Engineer Pune (India)
🏢 Devfixr
📍 India