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