MLOps Engineer (PyTorch)

ABOUT CLIENT

Our client is a leading research company specializing in technology innovation

JOB DESCRIPTION

Develop and manage training and inference pipelines using PyTorch
Create and maintain efficient tooling in Python and C++ to support the training process
Take ownership of the training codebase to ensure clarity, modularity, reproducibility, and performance
Plan and establish workflows for checkpointing, resuming, versioning, and tracking experiments
Streamline compute workloads for bare-metal environments including I/O, CPU/GPU utilization, and memory optimization
Address low-level networking issues, distributed training errors, and hardware bottlenecks
Establish and oversee ML environments including containers, package management, drivers, and runtime configs
Monitor and troubleshoot training jobs across multiple nodes and GPUs
Develop enduring systems designed for scalability, maintainability, and long-term usage

JOB REQUIREMENT

Strong proficiency in PyTorch, including DDP, mixed precision, and TorchScript
Proficient programming skills in C++ and Python
Sound understanding of computer science fundamentals including data structures, concurrency, and operating systems
Ability to debug and optimize bare-metal servers (Linux, kernel parameters, BIOS tuning)
Proficient understanding of networking, interconnects, and distributed training setups, including NCCL and MPI
Proven track record of creating dependable and reproducible pipelines for training and evaluation
Knowledge of job schedulers such as SLURM and custom batch runners, as well as monitoring tools
Experience with custom deployments (no cloud, local clusters, edge devices)
Contributions to PyTorch or open-source ML tooling
Familiarity with infrastructure-as-code tools like Ansible, Terraform, and Nix
Experience in setting up logging, observability, and alerting for training runs
A strong dedication to writing clean code and meticulous engineering practices

WHAT'S ON OFFER

Work remotely in an environment that promotes open-source collaboration
Enjoy 14 days of leave and unlimited sick days
Access to GPUs, AI credits, opportunities for fast career progression, and other perks.

CONTACT

PEGASI – IT Recruitment Consultancy | Email: recruit@pegasi.com.vn | Tel: +84 28 3622 8666
We are PEGASI – IT Recruitment Consultancy in Vietnam. If you are looking for new opportunity for your career path, kindly visit our website www.pegasi.com.vn for your reference. Thank you!

Job Summary

Company Type:

Product

Technical Skills:

Machine Learning, Devops

Location:

Others - Singapore

Salary:

Negotiation

Job ID:

J01855

Status:

Active