Founding AI Engineer
Remotefull-timePosted 10mo ago
About the role
Transform language models into real-world, high-impact product experiences by shaping AI model architecture, training, and deployment 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 learningModel training and fine-tuningFramework proficiencyDistributed training frameworksGPU optimizationSoftware engineering
Benefits
- Remote-first flexibility
- High-impact scope — your work becomes core infrastructure of a global consumer AI product
- 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 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