Founding AI/ML Research Engineer
full-timePosted 10mo ago
About the role
Join a self-funded, stealth AI group to shape core technical strategies for building a global consumer AI application, focusing on model development, training pipelines, and scalable deployment.
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 architecturesTraining large models (LLMs or vision models)PyTorch, JAX, or TensorFlowDistributed training frameworks (DeepSpeed, FSDP, Megatron, ZeRO, Ray)GPU optimization (memory efficiency, quantization, mixed precision)Software engineering for production systems
Benefits
- Remote-first flexibility
- Competitive compensation and performance-based bonuses
- Insurance coverage, flexible time off, and global travel insurance
- Opportunity to shape a new global AI product from zero
- Influence over technical direction, model architecture, and product strategy
- High-impact scope—core infrastructure of a global consumer AI product
- Backed by a profitable US$2B group, with startup speed
- Collaborate directly with founders and influence major decisions
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