Senior Distributed ML Engineer

Market insights

Stack

Apply

Upload your CV — we will connect you with the employer directly through our pool.

Description

We are seeking a senior distributed machine learning (ML) research developer to join our team working on a novel AI safety agenda. In this role, you will work closely with ML research scientists to solve difficult training and inference problems using very large models. Key Responsibilities Collaborate with researchers to accelerate research, model training and inference, and facilitate the use of large-scale models in distributed computing environments. Investigate performance bottlenecks, profile research experiment code, debug reported issues, and optimize the utilization of computing resources. Develop tools and libraries to simplify and orchestrate the use of distributed computing resources for research experiments. Establish, document, and maintain best practices for large-scale, distributed ML model development workflows. Skills and , computer science, computer engineering, software engineering) is required. An advanced degree (master’s or PhD) related to machine learning or distributed ML systems is preferred but not required if the candidate demonstrates exceptional abilities and experience. g. Megatron, DeepSpeed, HuggingFace Accelerate, FSDP, vLLM, and/or verl. Ability to collaborate effectively with cross-functional teams, document best practices, and stay updated with the latest advancements in ML and software development. , Ray, SLURM). g. PyTorch profiler, PyProf, NVIDIA

Employer contacts (email/phone/telegram) are hidden from the public preview — send your CV, and we will connect you directly.