Research Engineer, ML Systems & Foundation Models

Neuralk; Singapore; Hybrid; SGD 180,000 - 260,000; Full time

Benefits offered by this job

Equity (BSPCE)

State-of-the-art ML infra

Paid leave and time-off policy

Occasional remote work; offices in IP

Job summary

Neuralk is an early-stage AI startup building scalable predictive infrastructure for enterprise data science. We seek a Research Engineer to own pretraining, optimize inference latency, and write low-level GPU code to enable large-scale experimentation.

You will work at the intersection of systems and research, shaping architecture direction. The role emphasizes strong ML systems experience, Python/PyTorch expertise, and collaboration with a fast-moving team in a research-driven environment.

Qualifications

• Master’s or PhD in Computer Science, ML or related field.
• 5+ years of experience in ML systems focusing on training and inference optimization.
• Proficient in Python and PyTorch with solid software engineering practices.
• Hands-on experience writing CUDA kernels or using low-level GPU optimization libraries.
• Strong understanding of transformer architectures and scaling behavior.
• Experience with distributed training frameworks and cluster environments.
• Excellent English communication skills and ability to publish in ML venues.

Responsibilities

• Design and optimize transformer inference to reduce latency and scale production models.
• Build and maintain scalable pretraining infrastructure with distributed training and mixed precision.
• Develop CUDA kernels and low-level GPU code for bottleneck optimization.
• Collaborate with research to test architectures and reproduce experiments at scale.
• Develop internal tooling for experiments, evaluation, and benchmarking.
• Contribute to publications and engage with the ML community.

Skills

Python

PyTorch

CUDA kernels

Distributed training

Transformer models

English communication

Research publication

Inference serving

Education

Master’s or PhD in CS/ML

Tools

DeepSpeed

FSDP

SLURM

PyTorch

TensorRT

ONNX

Triton

cuBLAS

CUTLASS

Job description

Neuralk is an early-stage AI startup building scalable predictive infrastructure for enterprise data science. We seek a Research Engineer to own pretraining, optimize inference latency, and write low-level GPU code to enable large-scale experimentation.

You will work at the intersection of systems and research, shaping architecture direction. The role emphasizes strong ML systems experience, Python/PyTorch expertise, and collaboration with a fast-moving team in a research-driven environment.