Founding AI Engineer
Hybrid (Munich, Remote)full-timePosted 10mo ago
About the role
Building a global consumer AI application by transforming language models into impactful products; shaping technical direction including model selection, training, infrastructure, and architecture.
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 learningLarge model training / fine-tuningPyTorch, JAX, or TensorFlowDistributed training frameworks (DeepSpeed, FSDP, Megatron, ZeRO, Ray)GPU optimization (memory efficiency, quantization, mixed precision)Software engineering best practices
Benefits
- Remote-first flexibility
- High-impact scope—your work becomes core infrastructure of a global consumer AI product
- Competitive compensation and performance-based bonuses
- Backed by 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 6, 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