ML Research Scientist — Large-Model Training & Pipelines

SECOND TALENT SG PTE. LTD.; Singapore; On-site; SGD 80,000 - 120,000; Full time

Job summary

An innovative deep learning company in Singapore is seeking a Research Scientist to develop advanced training pipelines for deep learning research and collaborate with other scientists on architecture evolution. Strong skills in Python and experience with ML training using frameworks like PyTorch or JAX are crucial. This role offers an opportunity to work at the cutting edge of ML experimentation and infrastructure.

Qualifications

• Experience in training ML models at scale using frameworks like PyTorch or JAX.
• Proficiency in Python and familiarity with deep learning optimization internals.
• Understanding of GPU compute fundamentals and training-performance bottlenecks.

Responsibilities

• Develop and maintain high-performance training pipelines for deep learning research.
• Tune and debug large model training runs across clusters and GPU environments.
• Collaborate with other research scientists to evolve architecture and optimization techniques.

Skills

Training ML models at scale

Deep learning optimization

Python proficiency

Understanding GPU compute fundamentals

Clean and maintainable code

Tools

PyTorch

JAX

Job description

An innovative deep learning company in Singapore is seeking a Research Scientist to develop advanced training pipelines for deep learning research and collaborate with other scientists on architecture evolution. Strong skills in Python and experience with ML training using frameworks like PyTorch or JAX are crucial. This role offers an opportunity to work at the cutting edge of ML experimentation and infrastructure.