AI Research Scientist

ABOUT CLIENT

Our client is a leading research company specializing in technology innovation

JOB DESCRIPTION

Maintain a modular, high-quality PyTorch training codebase
Plan and establish training workflows for scaling, checkpointing, logging, and reproducibility
Introduce new concepts, troubleshoot training runs, and expedite iteration
Create and manage efficient data loading pipelines and training utilities
Ensure training jobs have the ability to scale across multiple GPUs and nodes (e.g., with DDP, NCCL)
Fine-tune model training for performance, stability, and hardware utilization
Uphold long-term code health: organize modules, enforce standards, write clean and testable code
Contribute to experiment tracking, reproducibility, and versioning infrastructure

JOB REQUIREMENT

Strong proficiency in PyTorch, including custom modules, loss functions, and distributed training
Demonstrated experience in training deep learning models in real-world research or production settings
Proficient engineering skills in Python, and optionally C++ for performance-critical components
Experience in working with large datasets, complex pipelines, and real-world debugging
Understanding of training dynamics and the ability to identify and resolve issues
Familiarity with job launchers, logging tools (e.g., Weights & Biases, TensorBoard), and checkpointing systems
Previous involvement with TorchScript, ONNX, or custom inference runtimes
Contributions to PyTorch or open-source ML tooling
Experience with transformer models, diffusion models, or large-scale vision/NLP tasks
Familiarity with batch schedulers (SLURM), cluster environments, and GPU resource management
Ability to collaborate closely with systems engineers or MLOps teams to ensure smooth integration

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:

Data Science

Location:

Others - Singapore

Salary:

Negotiation

Job ID:

J01854

Status:

Active

Related Job:

Senior DevOps (Data Platform)

Ho Chi Minh - Viet Nam

Digital Bank, Product

  • Devops
  • Spark

Managing workloads on EC2 clusters using DataBricks/EMR for efficient data processing Collaborating with stakeholders to implement a Data Mesh architecture for multiple closely related enterprise entities Utilizing Infrastructure as Code (IaC) tools for defining and managing data platform user access Implementing role-based access control (RBAC) mechanisms to enforce least privilege principles Collaborating with cross-functional teams to design, implement, and optimize data pipelines and workflows Utilizing distributed engines such as Spark for efficient data processing and analysis Establishing operational best practices for data warehousing tools Managing storage technologies to meet business requirements Troubleshooting and resolving platform-related issues Staying updated on emerging technologies and industry trends Documenting processes, configurations, and changes for comprehensive system documentation.

Negotiation

View details

Data Engineer

Ho Chi Minh - Viet Nam

Outsourcing

  • Data Engineering
  • Python

Enrich structured and unstructured data to create datasets. Improve data quality and integrity by developing validation tools to assess the effectiveness of data enrichment. Develop expertise in deep learning and machine learning applications, analyzing and understanding the underlying dynamics and behaviors in the data. Generate insights from the data and collaborate with the research team to create tradable signals. Develop utility tools to automate software development, testing, and deployment workflows. Provide technical support for global researchers, including diagnosing technical problems and proposing solutions to developers.

Negotiation

View details

QA Engineer (Data Testing)

Ho Chi Minh - Viet Nam

Outsourcing

  • Manual Test
  • Automation Test

Ensure the accuracy, completeness, and consistency of output through data pipeline, in line with business requirements and quality standards. Create and manage automation test cases using Python and/or Java to simplify data validation processes. Collaborate with data engineers, supporters, and other team members to identify and resolve data-related issues. Engage in discussions, provide updates, and cooperate with team members in English on a daily basis.

Negotiation

View details