BJAK is looking for a Machine Learning Engineer (LLM) to fine-tune state-of-the-art models, design evaluation frameworks, and bring AI features into production. Your work will ensure models are safe, trustworthy, and impactful at scale.
What You'll Do
- Fine-tune and adapt open-source models (LLaMA, Mistral, Gemma) using LoRA/QLoRA.
- Craft 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 (LoRA/QLoRA, SFT, distillation).
- Proficiency with PyTorch (preferred) 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
- LoRA, QLoRA, SFT, distillation
- RLHF
Team & Environment
Work closely with regional teams across product, engineering, operations, infrastructure and data.
Benefits & Compensation
- Housing rental subsidies.
- Quality company cafeteria.
- Overtime meals.
- Health, dental & vision insurance.
- Global travel insurance (for you & your dependents).
- Unlimited, flexible time off.
Work Mode
This is a hybrid role open to candidates in Malaysia, Thailand, Taiwan, or Japan.
BJAK is an equal opportunity employer.
