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