BJAK is hiring a Machine Learning Engineer specialized in Large Language Models. You will fine-tune state-of-the-art LLMs, design evaluation frameworks, and bring AI features into production to power how people interact with financial services. Your work will ensure models are safe, trustworthy, and impactful at scale.
What You'll Do
- Fine-tune and adapt open-source models like LLaMA, Mistral, and Gemma for Bjak's applications using methods like LoRA/QLoRA.
- Engineer prompts and curate datasets that reflect tone, brand voice, and safety.
- Build metrics pipelines for perplexity, toxicity, and relevance to ensure safe and high-quality outputs.
- Scale models into production with performance optimization and monitoring for drift.
- Partner with product, engineering, and design teams to launch user-facing AI features.
What We're Looking For
- Strong experience in transformers, deep learning, and fine-tuning methods like LoRA/QLoRA, SFT, or distillation.
- Proficiency with PyTorch or TensorFlow.
- Skilled in prompt engineering and dataset curation for alignment with tone, safety, and trust.
- Familiar with evaluation metrics: perplexity, toxicity, relevance.
- Strong software engineering foundations in algorithms, data structures, and clean code practices.
Nice to Have
- Prior work in text generation, moderation, or personalization.
- Experience with RLHF or reinforcement learning in LLMs.
- Contributions to open-source ML projects.
Technical Stack
- PyTorch
- TensorFlow
Team & Environment
The team has a flat structure and is a lean high-performance team.
Benefits & Compensation
- Competitive salary and performance-based bonuses.
- Flexibility in work arrangement.
- High-impact role with visibility across product, data, and engineering.
- Fast learning curve and accelerated growth.
- Cross-border exposure to regional product development.
Work Mode
This is a hybrid role based in Malaysia.
BJAK is an equal opportunity employer.
