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:

Close

Related Job:

OSS Software Engineer

Ho Chi Minh - Viet Nam

Product

  • C/C++

Collaborate with a small team of engineers to: Develop software primarily using C language Perform thorough software testing Deliver projects in phased releases Offer customer support for live operations Coordinate software development and support activities with the manager and the core development team based in California, USA

Negotiation

View details

Backend Engineer (Python/Kotlin)

Ho Chi Minh, Ha Noi - Viet Nam

Outsource

  • Python
  • Kotlin
  • AWS

Create and develop financial products on top of a core banking platform called Thought Machine Vault Create and develop event-driven microservices to enhance the functionality of the core banking platform Maintain and enhance service reliability through effective simulation, end-to-end, and performance tests Enhance SRE processes and offer production support for services Write high-quality, sustainable code using TDD Utilize Kubernetes and Docker for scheduling and running microservices Technology stack comprises mainly of Python and Kotlin/Java, with the flexibility to use the most suitable language for problem-solving Utilize PostgreSQL, Aurora, and S3 for persistence Harness the elastic AWS infrastructure Practice continuous integration and delivery Follow the approach of "you build it, you run it."

Negotiation

View details

Senior Full-stack Java Software Engineer

Ho Chi Minh - Viet Nam

Outsource

  • Java
  • Angular
  • Cloud

Create and uphold full-stack web applications using Java as the primary language, Spring Boot for the backend, and Angular (TypeScript) for the frontend. Develop and execute RESTful APIs with a focus on scalability, security, and performance. Engage in system design, technical discussions, and code review. Utilize AWS cloud services for the deployment and management of applications in production. Implement unit tests and integration tests to guarantee code quality. Contribute to Agile/Scrum teams alongside Product Owner, QA, and DevOps. Provide guidance and assistance to less experienced developers.

Negotiation

View details