AI for Science Research Associate — Data Pipelines & HPC

National University of Singapore; Singapore; On-site; SGD 70,000 - 100,000; Full time

Job summary

National University of Singapore (NUS) invites applications for a Research Associate in AI for Science. The role focuses on designing foundational layers for the Instrumentation Gym, connecting scientific data, ML representations, and HPC environments used by researchers at Kent Ridge Campus.

You will build modular Python interfaces, data pipelines, and scalable tokenisation workflows with a strong emphasis on reproducibility, performance on multi-node HPC, and collaboration with scientists

Qualifications

• Degree in a Computational, Engineering, or Physical-Science discipline.
• A Master's or Bachelor's with equivalent research-engineering experience.
• Experience building data pipelines and reproducible ML workflows end to end.

Responsibilities

• Data pipelines — ingestion, metadata schemas, and provenance tracking for electron microscopy, X-ray, and light microscopy datasets; format normalisation so one lab's deposit is usable by another.
• Model interfaces — modular Python APIs that let representation models, tokenizers, and downstream models be swapped and composed without rewriting the pipeline around them.
• Baseline tokenization and cartography pipelines, built collaboratively with ML researchers and domain scientists, as reference implementations others adapt for new domains.
• LLM- and RAG-assisted interfaces for dataset navigation, workflow discovery, and user interaction — retrieval over the Gym's own datasets, models, workflows, and deposited challenges.
• Forward-compatibility to HPC — components that scale from single-GPU teaching instances to multi-node; containerised, reproducible, scheduler-friendly deployment on NUS HPC.
• Onboarding and enablement — semi-regular sessions for new Gym users; helping instructors build workflows that generate teaching material from their own sources plus Gym content.

Skills

Python

PyTorch

Data pipelines

GPU compute

HPC

Education

Master's degree

Bachelor's degree with equivalent research-engineering experience

Tools

Docker

Apptainer

Slurm

HDF5

Zarr

Job description

National University of Singapore (NUS) invites applications for a Research Associate in AI for Science. The role focuses on designing foundational layers for the Instrumentation Gym, connecting scientific data, ML representations, and HPC environments used by researchers at Kent Ridge Campus.

You will build modular Python interfaces, data pipelines, and scalable tokenisation workflows with a strong emphasis on reproducibility, performance on multi-node HPC, and collaboration with scientists