Founding AI/ML Research Engineer
Remotefull-timePosted 10mo ago
About the role
A founding technical role focused on transforming language models into impactful products by shaping core model strategies, training pipelines, and infrastructure for a global consumer AI application.
Responsibilities
- Build end-to-end training pipelines: data → training → eval → inference
- Design new model architectures or adapt open-source frontier models
- Fine-tune models using state-of-the-art methods (LoRA/QLoRA, SFT, DPO, distillation)
- Architect scalable inference systems using vLLM / TensorRT-LLM / DeepSpeed
- Build data systems for high-quality synthetic and real-world training data
- Develop alignment, safety, and guardrail strategies
- Design evaluation frameworks across performance, robustness, safety, and bias
- Own deployment: GPU optimization, latency reduction, scaling policies
- Shape early product direction, experiment with new use cases, and build AI-powered experiences from zero
- Explore frontier techniques: retrieval-augmented training, mixture-of-experts, distillation, multi-agent orchestration, multimodal models
Qualifications
- Strong background in deep learning and transformer architectures
- Hands-on experience training or fine-tuning large models (LLMs or vision models)
- Proficiency with PyTorch, JAX, or TensorFlow
- Experience with distributed training frameworks (DeepSpeed, FSDP, Megatron, ZeRO, Ray)
- Strong software engineering skills — writing robust, production-grade systems
- Experience with GPU optimization: memory efficiency, quantization, mixed precision
- Comfortable owning ambiguous, zero-to-one technical problems end-to-end
Skills
Required:Deep Learning and Transformer ArchitecturesLarge Model Training and Fine-tuningProgramming with PyTorch, JAX, TensorFlowDistributed Training FrameworksGPU OptimizationSoftware Engineering SkillsOwnership of Technical Problems
Benefits
- Extreme ownership and autonomy from day one - you define and build key model systems.
- Founding-level influence over technical direction, model architecture, and product strategy.
- Remote-first flexibility
- High-impact scope—your work becomes core infrastructure of a global consumer AI product.
- Competitive compensation and performance-based bonuses
- Backing of a profitable US$2B group, with the speed of a startup
- Insurance coverage, flexible time off, and global travel insurance
- Opportunity to shape a new global AI product from zero
- A small, senior, high-performance team where you collaborate directly with founders and influence every major decision.
Last updated Dec 11, 2025. Interested? Apply on the company site.
About Bjak
No. 1 Insurance Platform in Malaysia
Malaysian's first choice for auto insurance and all types of insurance. Compare top providers, get instantly covered with roadtax fulfilled. Offers the best online insurance experience in the Malaysia.
insuranceauto insuranceonline insuranceMalaysia
More open roles at Bjak
- Product Designer (UI/UX)10mo agoHybrid (Malaysia) · Product & Design
- UI/UX Designer – Mobile10mo agoMalaysia · Product & Design
- Data Entry Admin10mo agoMalaysia · Operations
- Senior Lead, Events & Activation10mo agoMalaysia · Marketing
- Videographer10mo agoMalaysia · Marketing