Machine Learning Engineer (LLM)

BJAK


Date: 1 day ago
City: Petaling Jaya
Contract type: Full time
Transform Language Models into Real-World Applications

At Bjak, we’re building AI systems that reach millions across Southeast Asia. Our Machine Learning team is at the forefront of text generation, captioning, moderation, and chat-like interactions, deploying LLMs into real-world products that must be fast, safe, and aligned with our brand voice.

This role is based in Malaysia with a hybrid work arrangement - combining flexible remote work with in-office collaboration at our HQ. You’ll work closely with regional teams across product, engineering, and data to build and scale impactful AI solutions.

Why This Role Matters

We are scaling LLM capabilities that power how people interact with financial services. As a Machine Learning Engineer (LLM), you’ll fine-tune state-of-the-art models, design evaluation frameworks, and bring AI features into production. Your work ensures our models are not only intelligent, but also safe, trustworthy, and impactful at scale.

What You’ll Do

  • Fine-tune & Adapt - Use LoRA/QLoRA to optimize open-source models (LLaMA, Mistral, Gemma) for Bjak’s applications.
  • Engineer Prompts & Curate Data - Craft prompts and datasets that reflect tone, brand voice, and safety.
  • Evaluate Models – Build metrics pipelines for perplexity, toxicity, and relevance to ensure safe and high-quality outputs.
  • Deploy & Monitor – Scale models into production with performance optimization and monitoring for drift.
  • Collaborate & Deliver – Partner with product, engineering, and design teams to launch user-facing AI features.

You’ll Thrive Here If You...

  • Treat every problem as your own, regardless of scope or title.
  • Enjoy switching hats - builder, debugger, architect, and product thinker when needed.
  • Believe clarity comes from action - prototype, test, and iterate without waiting for perfect plans.
  • Stay calm and effective in startup chaos - shifting priorities and building from zero doesn’t faze you.
  • Bias for speed - you believe it’s better to deliver something valuable now than a perfect version much later.
  • See feedback and failure as part of growth - you’re here to level up.
  • Believe “done” beats “perfect,” but still care deeply about UX and code quality.
  • Show up with humility, hunger, and hustle, and lift others up as you go.

Requirements

  • 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.

What You’ll Get

  • Competitive salary and performance-based bonuses
  • Flexibility in work arrangement
  • High-impact role with visibility across product, data, and engineering
  • Flat structure - your work is seen, your ideas are heard
  • Fast learning curve and accelerated growth in a mission-driven company
  • Cross-border exposure to regional product development

Our Team & Culture

We’re a lean, high-performance team that moves fast and holds a high bar. Titles don’t matter here - output, integrity, and ownership do. Everyone is expected to roll up their sleeves, contribute meaningfully, and take initiative. We reward people who think like owners. We value speed, clarity, and relentless ownership. If you’re hungry to grow and care deeply about excellence, this is your kind of environment.

About Bjak

BJAK is Southeast Asia’s #1 insurance aggregator with 8M+ users, fully owned by its employees. Headquartered in Malaysia and operating in Thailand, Taiwan, and Japan, we help millions of users access transparent and affordable financial protection through Bjak.com. We simplify complex financial products through cutting-edge technologies, including APIs, automation, and AI, to build the next generation of intelligent financial systems.

If you're excited to build real-world AI systems and grow fast in a high-impact environment, we’d love to hear from you.

How to apply

To apply for this job you need to authorize on our website. If you don't have an account yet, please register.

Post a resume

Similar jobs

Senior Executive eCommerce Livestream

DKSH, Petaling Jaya
15 hours ago
Job SummaryThe Ecommerce Livestream Senior Executive must be dynamic and results-driven lead to the execution of our livestream commerce initiatives. This role is responsible for end-to-end livestream production planning, content strategy, pitching and on-air execution across digital platforms.Job ResponsibilitiesCommercial CollaborationCollaborate with commercial team to pitch livestream services to potential and existing clientsPresent livestream capabilities, value proposition, and past success casesLivestreamingResponsible...

Digital Marketer Cum Content Creator (Prefered Mandarin Speaker)

MR International Group, Petaling Jaya
MYR 3,000 - MYR 5,000 per month
17 hours ago
Job ResponsibilityPlan, execute, and manage paid media campaigns across social media platforms.Monitor, measure, report, and analyze campaign performance data to derive actionable insights and optimize advertising spend.Create and present comprehensive reports on campaign performance, including key metrics and recommendations for improvement.Monitor and manage campaign budgets, ensuring efficient allocation of resources to maximize results.Conduct A/B tests on ad creatives, targeting, and...

Procurement Analyst I (12 months contract)

Kimberly-Clark, Petaling Jaya
1 day ago
Job DescriptionProcurement Analyst I (12 months contract) - (Global Business Services)Your Job You’re not the person who will settle for just any role. Neither are we. Because we’re out to create Better Care for a Better World, and that takes a certain kind of person and teams who care about making a difference. Here, you’ll bring your professional expertise, talent,...