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

Working Policy:

Remote

Salary:

Negotiation

Job ID:

J01855

Status:

Active

Related Job:

Senior Systems Software Engineer

Ho Chi Minh, Ha Noi - Viet Nam

Computer Hardware

  • Devops

Contribute the development and maintenance of advanced machine learning software and frameworks, optimizing for performance and scalability. Enhance CI/CD pipelines to streamline the development, testing, and deployment of large-scale machine learning models. Implement and manage cloud infrastructure for continuous integration, delivery, and deployment, ensuring high availability and scalability. Collaborate with cross-functional teams, including engineering, QA, and research, to improve development workflows and enhance software delivery speed and quality. Troubleshoot and resolve complex issues related to software development, containerization, and cloud infrastructure in production environments. Write and maintain robust documentation for development and deployment processes. Communicate effectively with technical and non-technical stakeholders to set shared expectations and ensure visibility around the release and deployment process. Lead code reviews, testing, and debugging to ensure high-quality code and efficient workflows. Mentor and guide junior engineers, fostering professional growth and enhancing team capabilities.

Negotiation

View details

Senior SWQA Test Developer Engineer, SDET

Ho Chi Minh - Viet Nam

Computer Hardware

  • SDET
  • Automation Test
  • Python
  • AI

Review product requirements and develop test matrix. Build test plan, design test case, execute and report test progress, bugs, and results to management. Automate test cases and assist in the architecture, crafting and implementing of test frameworks. Manage bug lifecycle and co-work with inter-groups to drive for solutions. In-house repro and verify customer issues/fixes.

Negotiation

View details

SWQA Development Engineer

Ho Chi Minh - Viet Nam

Computer Hardware

  • SDET
  • Automation Test
  • Python
  • AI

Review product requirements and develop test matrix. Build test plan, design test case, execute and report test progress, bugs, and results to management. Automate test cases and assist in the architecture, crafting and implementing of test frameworks. Manage bug lifecycle and co-work with inter-groups to drive for solutions. In-house repro and verify customer issues/fixes.

Negotiation

View details