Graduate Model Infrastructure Engineer - LLM & GPU Systems

TIKTOK PTE. LTD.; Singapore; On-site; SGD 60,000 - 80,000; Full time

Job summary

A leading technology firm in Singapore is seeking new graduates to join their Recommendation Architecture team. This role involves optimizing large language models and building AI infrastructure on GPU platforms. Candidates will collaborate on extensive projects and should have solid programming skills in C++ and familiarity with GPU architecture. The firm offers competitive compensation, flexibility, and a positive team atmosphere. Apply early to secure your position in an innovative and dynamic environment.

Benefits

Yoga and fitness

Stock options

Positive team atmosphere

Career growth opportunity

Paid leave

Flat organization

Competitive compensation

Flexible hours

Qualifications

• Final year or recent graduate with a background in Computer Science.
• Solid programming skills in C++/CUDA/Python.
• Familiarity with GPU architecture and distributed training is desirable.

Responsibilities

• Optimize model performance and memory efficiency on GPU systems.
• Collaborate with teams to deploy high-throughput training pipelines.
• Analyze system performance through GPU profiling and throughput tuning.

Skills

C++

CUDA

Python

GPU architecture

Education

Bachelor's degree in Computer Science or related field

Tools

DeepSpeed

TensorFlow

PyTorch

Job description

A leading technology firm in Singapore is seeking new graduates to join their Recommendation Architecture team. This role involves optimizing large language models and building AI infrastructure on GPU platforms. Candidates will collaborate on extensive projects and should have solid programming skills in C++ and familiarity with GPU architecture. The firm offers competitive compensation, flexibility, and a positive team atmosphere. Apply early to secure your position in an innovative and dynamic environment.