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.
- 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.
📌 Machine Learning Engineer (Pune)
🏢 Devfixr
📍 Pune
Reply to this offer
Impress this employer describing Your skills and abilities, fill out the form below and leave Your personal touch in the presentation letter.