Founding ML Engineer — Shape the Core AI Platform

Bjak; Remote; MYR 250,000 - 300,000; Full time

Job summary

An AI-focused company in Selangor is looking for a founding technical role to shape their new global consumer AI application. You will work on model selection, training strategies, and pipeline design, influencing the functionality and strategy of the initial product. Candidates should have a strong background in deep learning including hands-on experience with large models and proficiency in frameworks such as PyTorch or TensorFlow. This role offers competitive compensation and

Benefits

Remote-first flexibility

Competitive compensation

Insurance coverage

Qualifications

• Strong background in deep learning and transformer architectures required.
• Experience training or fine-tuning large models essential.
• Proficient with PyTorch, JAX, or TensorFlow expected.

Responsibilities

• Build end-to-end training pipelines including data and inference systems.
• Design and adapt model architectures.
• Own technical problems from concept to deployment.

Skills

Deep learning

Transformer architectures

Hands-on model training

Software engineering

GPU optimization

Tools

PyTorch

TensorFlow

JAX

DeepSpeed

Job description

An AI-focused company in Selangor is looking for a founding technical role to shape their new global consumer AI application. You will work on model selection, training strategies, and pipeline design, influencing the functionality and strategy of the initial product. Candidates should have a strong background in deep learning including hands-on experience with large models and proficiency in frameworks such as PyTorch or TensorFlow. This role offers competitive compensation and