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:

Engagement Engineer

Ho Chi Minh - Viet Nam

Product, AI Application Platform

  • Project Management
  • Account Management
  • AI

Drive customer success by strategically managing relationships, acting as a key advocate, and aligning our product capabilities with customer goals while proactively addressing project risks. Stay updated on AI innovation by participating in the design of AI systems for industrial applications and keeping up with the latest trends in AI, machine learning, and data analytics. Foster customer collaboration to clearly define problem statements, develop impactful solutions, translate needs into actionable tasks, and oversee project lifecycles, including roadmap planning, execution, and quality assurance. Lead and collaborate with cross-functional technical teams to efficiently develop and deliver solutions within specified timelines. Provide customers with tailored technical solutions and training to enhance the impact and value of the solutions offered.

Negotiation

View details

API Integration Engineer (Java/Python/Golang)

Ho Chi Minh - Viet Nam

Offshore

  • Java
  • Python

Main responsibilities include integrating using Java/Python/Golang, REST, SOAP APIs, and Identity service. Addressing complex technical and business challenges and staying updated on new technology and frameworks. Collaborating with a team to take accountability for the features you manage. Handling the complete product life cycle—from design and development to testing, deployment, monitoring, and enhancement.

Negotiation

View details

Platform Reliability Engineer

Ho Chi Minh - Viet Nam

Outsource

Maintain production reliability of the Linux-based research and trading platform within a globally distributed engineering team. Respond quickly to production infrastructure issues. Comprehend internal client needs and effectively communicate them to regional and global leadership. Identify risks, develop contingency plans, and implement solutions to mitigate them. Enhance the observability platform to monitor the performance and health of critical computing environments. Take part in occasional on-call rotations and support on-call staff during their shifts. Contribute to organizational knowledge through documentation, education, and writing maintainable code.

Negotiation

View details