Remote Founding AI/ML Systems Architect

Bjak; Petaling Jaya; Remote; MYR 150,000 - 200,000; Full time

Job summary

A self-funded AI group is seeking a founding technical figure to lead the development of a new global consumer AI application. In this role, you'll guide the model selection, training strategies, and architecture for high-impact AI systems. You'll have the unique opportunity to create end-to-end training pipelines, design innovative architectures, and influence the product’s direction from the ground up. This position offers competitive compensation, remote flexibility, and a chance to work closely with elite team members to tackle groundbreaking challenges.

Benefits

Competitive compensation

Performance-based bonuses

Insurance coverage

Flexible time off

Global travel insurance

Qualifications

• Strong background in deep learning and transformer architectures.
• Hands-on experience training or fine-tuning large models.
• Comfortable owning ambiguous technical problems end-to-end.

Responsibilities

• Build end-to-end training pipelines: data → training → eval → inference.
• Design new model architectures or adapt existing ones.
• Architect scalable inference systems using cutting-edge technologies.

Skills

Deep learning

Transformer architectures

PyTorch

JAX

TensorFlow

GPU optimization

Tools

DeepSpeed

FSDP

Megatron

ZeRO

Ray

Job description

A self-funded AI group is seeking a founding technical figure to lead the development of a new global consumer AI application. In this role, you'll guide the model selection, training strategies, and architecture for high-impact AI systems. You'll have the unique opportunity to create end-to-end training pipelines, design innovative architectures, and influence the product’s direction from the ground up. This position offers competitive compensation, remote flexibility, and a chance to work closely with elite team members to tackle groundbreaking challenges.