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 Signal Processing Engineer

Others - Viet Nam

Outsource

  • Python

Design and improve rPPG/TOI pipelines using RGB/IR video with motion/illumination compensation. Implement multi-stage preprocessing, denoising, and quality scoring. Examples: adaptive filtering, ICA/PCA variants, color-space transforms, skin ROI stabilization, signal confidence metrics. o Build cross-device normalization strategies and error-bounded estimators. Define biomarker-level acceptance criteria and failure modes for consumer-grade capture. Partner with iOS and ML teams to integrate algorithms into on-device or hybrid pipelines. Produce technical documentation suitable for regulatory-risk positioning and Apple review support. Benchmark report across device models, skin tones, lighting, and motion conditions. Biomarker feature specification sheet with recommended thresholds and confidence bands. A/B results showing improvements in stability, missingness, and downstream inference performance.

Negotiation

View details

Senior Mobile App Development Engineer

Others - Viet Nam

Outsource

  • iOS

Develop and enhance capture flows using TrueDepth, ARKit, AVFoundation, and CoreMotion. Convert algorithm requirements into reliable on-device data collection and frame synchronization. Create user-friendly quality control interface and capture-state instrumentation, including tasks such as lighting checks, face alignment, motion thresholds, occlusion detection, and retake prompts. Improve latency, thermal behavior, and battery usage for 10-second diagnostic-style capture. Ensure data privacy and compliance with privacy regulations in telemetry. Create technical documentation for the use of specific APIs and data handling for Apple's internal use. Build a modular capture SDK layer with feature flags and logging capabilities. Develop a cross-device performance matrix and tuning guide. Integrate deterministic quality control gates to minimize noisy biomarker outputs in the capture process.

Negotiation

View details

Senior Software Engineer, Machine Learning

Ho Chi Minh - Viet Nam

Product

  • Machine Learning

Building and maintaining high-scale ML pipelines using big data technologies (Spark, BigQuery, Delta Lake), writing reliable, testable, and maintainable production-ready code. Leveraging modern ML platforms like Vertex AI or Databricks to orchestrate workflows and optimizing models for both batch/offline training and real-time/streaming inference with a focus on low latency. Gaining a deep understanding of specific domain challenges (e.g., fraud detection, location spoofing) to engineer better features and network-level intelligence that go beyond algorithms. Championing best practices in software engineering, including CI/CD, automated testing, code reviews, and documentation to ensure ML systems are built to software engineering standards, not just research scripts. Acting as a technical lead for projects, defining milestones and ensuring on-time delivery, mentoring Data Scientists on engineering best practices, and collaborating with other engineers to integrate models into the wider platform.

Negotiation

View details