Founding AI Engineer
full-timePosted 10mo ago
About the role
A founding technical role focused on building a global consumer AI application by shaping core model and system architecture, training pipelines, and experimental approaches.
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 & transformer architecturesModel training & fine-tuningPyTorch / JAX / TensorFlowDistributed training frameworksGPU optimizationSoftware engineering
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
- A small, senior, high-performance team working directly with founders
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